Spaces:
Sleeping
Sleeping
patrickvonplaten
commited on
Commit
•
97f4a6d
1
Parent(s):
a01639c
fix to numpy
Browse filesFormer-commit-id: b9851783e5400900a5de8ad43cb6f407f2cd1e4d
- scripts/txt2img.py +1 -1
scripts/txt2img.py
CHANGED
@@ -266,7 +266,7 @@ def main():
|
|
266 |
|
267 |
x_samples_ddim = model.decode_first_stage(samples_ddim)
|
268 |
x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
|
269 |
-
x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1)
|
270 |
|
271 |
x_image = x_samples_ddim
|
272 |
safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt")
|
|
|
266 |
|
267 |
x_samples_ddim = model.decode_first_stage(samples_ddim)
|
268 |
x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
|
269 |
+
x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
|
270 |
|
271 |
x_image = x_samples_ddim
|
272 |
safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt")
|