osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero commited on Aug 3, 2021

Commit

af0640a

·

1 Parent(s): 6ef6b1d

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -37,7 +37,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")