Spaces:
Runtime error
Runtime error
Commit
·
9202e10
1
Parent(s):
9df27df
Update utils.py
Browse files
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 |
|