tolgacangoz commited on
Commit
564a57e
·
verified ·
1 Parent(s): 7e9a5d3

Upload matryoshka.py

Browse files
Files changed (1) hide show
  1. 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
- # 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)
 
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)