goooo

configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml

@@ -110,11 +110,12 @@ lightning:
     image_logger:
       target: main.ImageLogger
       params:
-        batch_frequency: 50
+        batch_frequency: 5000
         max_images: 4
         increase_log_steps: False
         log_first_step: False
         log_images_kwargs:
+          use_ema_scope: False
           inpaint: False
           plot_progressive_rows: False
           plot_diffusion_rows: False

ldm/models/diffusion/ddpm.py

@@ -1194,7 +1194,9 @@ class LatentDiffusion(DDPM):
             if isinstance(xc, dict) or isinstance(xc, list):
                 c = self.get_learned_conditioning(xc)
             else:
-                c = self.get_learned_conditioning(xc.to(self.device))
+                if hasattr(xc, "to"):
+                    xc = xc.to(self.device)
+                c = self.get_learned_conditioning(xc)
         else:
             # todo: get null label from cond_stage_model
             raise NotImplementedError()