neural-os

Runtime error

yuntian-deng commited on Nov 21, 2024

Commit

da25b0f

1 Parent(s): 03c7138

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

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

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