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

Upload matryoshka.py

Browse files
Files changed (1) hide show
  1. unet/nesting_level_0/matryoshka.py +2 -2
unet/nesting_level_0/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