yuntian-deng commited on
Commit
dbf3e56
·
1 Parent(s): 620d901

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -41,7 +41,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
41
  c = model.enc_concat_seq(c, c_dict, 'c_concat')
42
  if pos_map is not None:
43
  print (pos_map.shape, c['c_concat'].shape)
44
- #c['c_concat'] = torch.cat([c['c_concat'][:, :, :, :], pos_map.to(c['c_concat'].device).unsqueeze(0)], dim=1)
45
 
46
  print ('sleeping')
47
  #time.sleep(120)
 
41
  c = model.enc_concat_seq(c, c_dict, 'c_concat')
42
  if pos_map is not None:
43
  print (pos_map.shape, c['c_concat'].shape)
44
+ c['c_concat'] = torch.cat([c['c_concat'][:, :, :, :], pos_map.to(c['c_concat'].device).unsqueeze(0)], dim=1)
45
 
46
  print ('sleeping')
47
  #time.sleep(120)