yuntian-deng commited on
Commit
7c4f4db
·
1 Parent(s): 80bed0e

Update utils.py

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