Hugging Face's logo

Spaces:
yuntian-group
/
neural-os
Runtime error

App Files Community
yuntian-deng commited on
Commit
abac87c
·
1 Parent(s): ddf2486

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -55,7 +55,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
55
  print ('sleeping')
56
  #time.sleep(120)
57
  print ('finished sleeping')
58
- DDPM = True
59
  if DDPM:
60
  samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 64, 64], return_intermediates=False, verbose=True)
61
  else:
 
55
  print ('sleeping')
56
  #time.sleep(120)
57
  print ('finished sleeping')
58
+ DDPM = False
59
  if DDPM:
60
  samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 64, 64], return_intermediates=False, verbose=True)
61
  else: