osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero HF staff commited on Aug 16, 2021

Commit

af7ada6

•

1 Parent(s): 8298f98

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -34,6 +34,6 @@ class PreTrainedPipeline():
         # 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)
         return img

         # 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)
         return img