tolgacangoz commited on
Commit
f69ddd7
1 Parent(s): 7c9ad15

Upload matryoshka.py

Browse files
Files changed (1) hide show
  1. scheduler/matryoshka.py +2 -2
scheduler/matryoshka.py CHANGED
@@ -4633,8 +4633,8 @@ class MatryoshkaPipeline(
4633
  image = latents
4634
 
4635
  if self.scheduler.scales is not None:
4636
- for i, (img, scale) in enumerate(zip(image, self.scheduler.scales)):
4637
- image[i] = self.image_processor.postprocess(img * scale, output_type=output_type)[0]
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