da03 commited on
Commit
9c3326f
·
1 Parent(s): 18d5c14
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -97,7 +97,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
97
  data_std = 6.78
98
  data_min = -27.681446075439453
99
  data_max = 30.854148864746094
100
- x_samples_ddim = x_samples_ddim * data_std + data_mean
101
  x_samples_ddim = model.decode_first_stage(x_samples_ddim)
102
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)
103
  #x_samples_ddim = model.decode_first_stage(x_samples_ddim)
 
97
  data_std = 6.78
98
  data_min = -27.681446075439453
99
  data_max = 30.854148864746094
100
+ x_samples_ddim = samples_ddim * data_std + data_mean
101
  x_samples_ddim = model.decode_first_stage(x_samples_ddim)
102
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)
103
  #x_samples_ddim = model.decode_first_stage(x_samples_ddim)