without ema

config.yaml

@@ -18,7 +18,7 @@ model:
     scale_factor: 0.18215
     only_mid_control: False
     sd_locked: True
-    use_ema: True #False #True #False
+    use_ema: False #True #False #True #False
 
     control_stage_config:
       target: cldm.cldm.ControlNet

ldm/models/diffusion/ddpm.py

@@ -626,7 +626,9 @@ class LatentDiffusion(DDPM):
             self.scale_factor = scale_factor
         else:
             self.register_buffer('scale_factor', torch.tensor(scale_factor))
+        print("instantiate first stage model")
         self.instantiate_first_stage(first_stage_config)
+        print("instantiate cond stage model")
         self.instantiate_cond_stage(cond_stage_config)
         self.cond_stage_forward = cond_stage_forward
         self.clip_denoised = False