yuntian-deng commited on
Commit
9202e10
·
1 Parent(s): 9df27df

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -49,7 +49,8 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
49
  eta=0)
50
 
51
  x_samples_ddim = model.decode_first_stage(samples_ddim)
52
- x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
 
53
 
54
  return x_samples_ddim.squeeze(0).cpu().numpy()
55
 
 
49
  eta=0)
50
 
51
  x_samples_ddim = model.decode_first_stage(samples_ddim)
52
+ #x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
53
+ x_samples_ddim = torch.clamp(x_samples_ddim, min=-1.0, max=1.0)
54
 
55
  return x_samples_ddim.squeeze(0).cpu().numpy()
56