Hugging Face's logo

Spaces:
yuntian-group
/
neural-os
Runtime error

App Files Community
yuntian-deng commited on
Commit
ad1b655
·
1 Parent(s): cd40774

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -56,7 +56,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
56
  # unconditional_conditioning=uc,
57
  # eta=0)
58
 
59
- x_samples_ddim = model.decode_first_stage(samples_ddim)
60
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)
61
  #x_samples_ddim = model.decode_first_stage(x_samples_ddim)
62
  #x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
 
56
  # unconditional_conditioning=uc,
57
  # eta=0)
58
 
59
+ x_samples_ddim = samples_ddim #model.decode_first_stage(samples_ddim)
60
  #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)
61
  #x_samples_ddim = model.decode_first_stage(x_samples_ddim)
62
  #x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)