yuntian-deng commited on
Commit
c520c64
·
1 Parent(s): 53ee11a

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -32,9 +32,9 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
32
  sampler = DDIMSampler(model)
33
 
34
  with torch.no_grad():
35
- u_dict = {'c_crossattn': "", 'c_concat': image_sequence}
36
- uc = model.get_learned_conditioning(u_dict)
37
- uc = model.enc_concat_seq(uc, u_dict, 'c_concat')
38
 
39
  c_dict = {'c_crossattn': prompt, 'c_concat': image_sequence}
40
  c = model.get_learned_conditioning(c_dict)
 
32
  sampler = DDIMSampler(model)
33
 
34
  with torch.no_grad():
35
+ #u_dict = {'c_crossattn': "", 'c_concat': image_sequence}
36
+ #uc = model.get_learned_conditioning(u_dict)
37
+ #uc = model.enc_concat_seq(uc, u_dict, 'c_concat')
38
 
39
  c_dict = {'c_crossattn': prompt, 'c_concat': image_sequence}
40
  c = model.get_learned_conditioning(c_dict)