neural-os

Runtime error

yuntian-deng commited on Nov 18, 2024

Commit

0abd296

1 Parent(s): a10a91e

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -53,7 +53,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         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:

         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: