da03 commited on
Commit
8c27ef5
·
1 Parent(s): 6ee36ca
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -108,8 +108,8 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
108
  data_min = -27.681446075439453
109
  data_max = 30.854148864746094
110
  x_samples_ddim = samples_ddim
111
- x_samples_ddim = x_samples_ddim * data_std + data_mean
112
  x_samples_ddim_feedback = x_samples_ddim
 
113
  x_samples_ddim = model.decode_first_stage(x_samples_ddim)
114
  print ('dfsf3')
115
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)
 
108
  data_min = -27.681446075439453
109
  data_max = 30.854148864746094
110
  x_samples_ddim = samples_ddim
 
111
  x_samples_ddim_feedback = x_samples_ddim
112
+ x_samples_ddim = x_samples_ddim * data_std + data_mean
113
  x_samples_ddim = model.decode_first_stage(x_samples_ddim)
114
  print ('dfsf3')
115
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)