osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero commited on Aug 16, 2021

Commit

7c9b1b1

·

1 Parent(s): af7ada6

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -32,7 +32,7 @@ class PreTrainedPipeline():
         with torch.no_grad():
             output = self.model(noise_vector, class_vector, self.truncation)
-        # Ugly hack, I'm sure there's a better way of doing what I want.
         img = output[0]
         img = (img + 1) / 2.0
         img = transforms.ToPILImage()(img)

         with torch.no_grad():
             output = self.model(noise_vector, class_vector, self.truncation)
+        # Scale image
         img = output[0]
         img = (img + 1) / 2.0
         img = transforms.ToPILImage()(img)