osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero commited on Aug 3, 2021

Commit

f52bcf2

·

1 Parent(s): 1b17997

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -32,6 +32,6 @@ class PreTrainedPipeline():
         class_vector = torch.from_numpy(class_vector)
         with torch.no_grad():
-            output = self.model(noise_vector, class_vector, truncation)
         return transforms.ToPILImage()(output[0])

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