From 4087dbfbb660856e2a67afc719db62ebc6e6becf Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 9 Oct 2023 15:36:27 +0200 Subject: [PATCH] step by step debug --- src/diffusers/models/controlnet.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/models/controlnet.py b/src/diffusers/models/controlnet.py index f32e789079..afddb04e51 100644 --- a/src/diffusers/models/controlnet.py +++ b/src/diffusers/models/controlnet.py @@ -722,6 +722,7 @@ class ControlNetModel( timesteps = timesteps.expand(sample.shape[0]) t_emb = self.time_proj(timesteps) + print(f"t_emb: {t_emb}") # timesteps does not contain any weights and will always return f32 tensors # but time_embedding might actually be running in fp16. so we need to cast here. @@ -729,6 +730,8 @@ class ControlNetModel( t_emb = t_emb.to(dtype=sample.dtype) emb = self.time_embedding(t_emb, timestep_cond) + print(f"emb: {emb}") + aug_emb = None if self.class_embedding is not None: