neural-os

Runtime error

da03 commited on Nov 18, 2024

Commit

8f4e2db

1 Parent(s): 313bb52

.

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -55,7 +55,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         print ('sleeping')
         #time.sleep(120)
         print ('finished sleeping')
-        DDPM = False
         if DDPM:
             samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 64, 64], return_intermediates=False, verbose=True)
         else:

         print ('sleeping')
         #time.sleep(120)
         print ('finished sleeping')
+        DDPM = True
         if DDPM:
             samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 64, 64], return_intermediates=False, verbose=True)
         else: