da03 commited on
Commit
aab7be4
·
1 Parent(s): 4a61676
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -37,14 +37,14 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
37
  #uc = model.get_learned_conditioning(u_dict)
38
  #uc = model.enc_concat_seq(uc, u_dict, 'c_concat')
39
 
40
- c_dict = {'c_crossattn': prompt, 'c_concat': image_sequence}
41
 
42
  model.eval()
43
  #c = model.get_learned_conditioning(c_dict)
44
  #print (c['c_crossattn'].shape)
45
  #print (c['c_crossattn'][0])
46
  print (prompt)
47
- c = {}
48
  #c = model.enc_concat_seq(c, c_dict, 'c_concat')
49
  # Zero out the corresponding subtensors in c_concat for padding images
50
  #padding_mask = torch.isclose(image_sequence, torch.tensor(-1.0), rtol=1e-5, atol=1e-5).all(dim=(1, 2, 3)).unsqueeze(0)
 
37
  #uc = model.get_learned_conditioning(u_dict)
38
  #uc = model.enc_concat_seq(uc, u_dict, 'c_concat')
39
 
40
+ #c_dict = {'c_crossattn': prompt, 'c_concat': image_sequence}
41
 
42
  model.eval()
43
  #c = model.get_learned_conditioning(c_dict)
44
  #print (c['c_crossattn'].shape)
45
  #print (c['c_crossattn'][0])
46
  print (prompt)
47
+ c = {'c_concat': image_sequence}
48
  #c = model.enc_concat_seq(c, c_dict, 'c_concat')
49
  # Zero out the corresponding subtensors in c_concat for padding images
50
  #padding_mask = torch.isclose(image_sequence, torch.tensor(-1.0), rtol=1e-5, atol=1e-5).all(dim=(1, 2, 3)).unsqueeze(0)