da03 commited on
Commit
9caafdc
·
1 Parent(s): fe752b2
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -82,7 +82,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
82
  if DDPM:
83
  samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 48, 64], return_intermediates=False, verbose=True)
84
  else:
85
- samples_ddim, _ = sampler.sample(S=4,
86
  conditioning=c,
87
  batch_size=1,
88
  shape=[4, 48, 64],
 
82
  if DDPM:
83
  samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 48, 64], return_intermediates=False, verbose=True)
84
  else:
85
+ samples_ddim, _ = sampler.sample(S=8,
86
  conditioning=c,
87
  batch_size=1,
88
  shape=[4, 48, 64],