Update utils.py

utils.py

@@ -58,12 +58,12 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         DDPM = False
         DDPM = False
         if DDPM:
-            samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 64, 48], return_intermediates=False, verbose=True)
+            samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 48, 64], return_intermediates=False, verbose=True)
         else:
             samples_ddim, _ = sampler.sample(S=8,
                                          conditioning=c,
                                          batch_size=1,
-                                         shape=[4, 64, 48],
+                                         shape=[4, 48, 64],
                                          verbose=False)
         #                                 unconditional_guidance_scale=5.0,
         #                                 unconditional_conditioning=uc,