neural-os

Runtime error

da03 commited on Nov 18, 2024

Commit

176343f

2 Parent(s): 50eea75 0abd296

Merge branch 'main' of hf.co:spaces/yuntian-deng/computer into main

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: