Upload matryoshka.py
Browse files- matryoshka.py +2 -2
matryoshka.py
CHANGED
|
@@ -4633,8 +4633,8 @@ class MatryoshkaPipeline(
|
|
| 4633 |
image = latents
|
| 4634 |
|
| 4635 |
if self.scheduler.scales is not None:
|
| 4636 |
-
for i,
|
| 4637 |
-
image[i] = self.image_processor.postprocess(img
|
| 4638 |
else:
|
| 4639 |
image = self.image_processor.postprocess(image, output_type=output_type)
|
| 4640 |
|
|
|
|
| 4633 |
image = latents
|
| 4634 |
|
| 4635 |
if self.scheduler.scales is not None:
|
| 4636 |
+
for i, img in enumerate(image):
|
| 4637 |
+
image[i] = self.image_processor.postprocess(img, output_type=output_type)[0]
|
| 4638 |
else:
|
| 4639 |
image = self.image_processor.postprocess(image, output_type=output_type)
|
| 4640 |
|