Update utils.py

utils.py

@@ -46,12 +46,12 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         print ('sleeping')
         #time.sleep(120)
         print ('finished sleeping')
-        samples_ddim = model.p_sample_loop(cond=c, shape=[1, 3, 256, 256], return_intermediates=False, verbose=True)
-        #samples_ddim, _ = sampler.sample(S=999,
-        #                                 conditioning=c,
-        #                                 batch_size=1,
-        #                                 shape=[3, 64, 64],
-        #                                 verbose=False,
+        #samples_ddim = model.p_sample_loop(cond=c, shape=[1, 3, 256, 256], return_intermediates=False, verbose=True)
+        samples_ddim, _ = sampler.sample(S=4,
+                                         conditioning=c,
+                                         batch_size=1,
+                                         shape=[3, 256, 256],
+                                         verbose=False)
         #                                 unconditional_guidance_scale=5.0,
         #                                 unconditional_conditioning=uc,
         #                                 eta=0)