osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero HF staff commited on Aug 3, 2021

Commit

4a8794e

•

1 Parent(s): bc34c30

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -35,7 +35,7 @@ class PreTrainedPipeline():
         with torch.no_grad():
             output = self.model(noise_vector, class_vector, self.truncation)
-        img = transforms.ToPILImage()(output[0])
         buf = io.BytesIO()
         img.save(buf, format="JPEG")

         with torch.no_grad():
             output = self.model(noise_vector, class_vector, self.truncation)
+        img = transforms.ToPILImage(output[0])
         buf = io.BytesIO()
         img.save(buf, format="JPEG")