osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

Narsil HF staff commited on Aug 16, 2021

Commit

66d0a2c

·

1 Parent(s): 55fd70d

Fixing scaling ?

Files changed (1) hide show

pipeline.py +4 -1

pipeline.py CHANGED Viewed

@@ -33,5 +33,8 @@ class PreTrainedPipeline():
             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 = transforms.ToPILImage()(output[0])
         return img

             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.transpose((0, 2, 3, 1))
+        img = (img + 1) / 2.0)
+        img = transforms.ToPILImage()(img)
         return img