Upload matryoshka.py
Browse files- scheduler/matryoshka.py +1 -1
scheduler/matryoshka.py
CHANGED
@@ -4660,7 +4660,7 @@ class MatryoshkaPipeline(
|
|
4660 |
for i in range(len(image))
|
4661 |
]
|
4662 |
for i, (img, scale) in enumerate(zip(image, scales)):
|
4663 |
-
|
4664 |
img = torch.clamp(img * 0.5 + 0.5, min=0, max=1).cpu()
|
4665 |
img = img.squeeze(0).permute(1, 2, 0).numpy()
|
4666 |
# img = self.image_processor.pt_to_numpy(img)
|
|
|
4660 |
for i in range(len(image))
|
4661 |
]
|
4662 |
for i, (img, scale) in enumerate(zip(image, scales)):
|
4663 |
+
img = torch.clip(img * scale, -1, 1)
|
4664 |
img = torch.clamp(img * 0.5 + 0.5, min=0, max=1).cpu()
|
4665 |
img = img.squeeze(0).permute(1, 2, 0).numpy()
|
4666 |
# img = self.image_processor.pt_to_numpy(img)
|