yuntian-deng commited on
Commit
0bea37e
·
1 Parent(s): 2c28657

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -5
utils.py CHANGED
@@ -39,14 +39,14 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
39
  c = model.get_learned_conditioning(c_dict)
40
  c = model.enc_concat_seq(c, c_dict, 'c_concat')
41
 
42
- samples_ddim, _ = sampler.sample(S=50,
43
  conditioning=c,
44
  batch_size=1,
45
  shape=[3, 64, 64],
46
- verbose=False,
47
- unconditional_guidance_scale=5.0,
48
- unconditional_conditioning=uc,
49
- eta=0)
50
 
51
  x_samples_ddim = model.decode_first_stage(samples_ddim)
52
  #x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
 
39
  c = model.get_learned_conditioning(c_dict)
40
  c = model.enc_concat_seq(c, c_dict, 'c_concat')
41
 
42
+ samples_ddim, _ = sampler.sample(S=999,
43
  conditioning=c,
44
  batch_size=1,
45
  shape=[3, 64, 64],
46
+ verbose=False, eta=1)
47
+ #unconditional_guidance_scale=5.0,
48
+ #unconditional_conditioning=uc,
49
+ #eta=0)
50
 
51
  x_samples_ddim = model.decode_first_stage(samples_ddim)
52
  #x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)