tolgacangoz
commited on
Commit
•
9964995
1
Parent(s):
67e1373
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 |
|