osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero HF staff commited on Aug 3, 2021

Commit

4afceb6

•

1 Parent(s): 740f729

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -31,7 +31,7 @@ class PreTrainedPipeline():
         noise_vector = torch.from_numpy(noise_vector)
         class_vector = torch.from_numpy(class_vector)
         with torch.no_grad():
-            output = self.model(noise_vector, class_vector, truncation)
         img = transforms.ToPILImage()(output[0])
         buffer = BytesIO()

         noise_vector = torch.from_numpy(noise_vector)
         class_vector = torch.from_numpy(class_vector)
         with torch.no_grad():
+            output = self.model(noise_vector, class_vector, self.truncation)
         img = transforms.ToPILImage()(output[0])
         buffer = BytesIO()