Update pipeline.py

pipeline.py

@@ -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)
+            output = self.model(noise_vector, class_vector, self.truncation)
 
         img = transforms.ToPILImage()(output[0])
         buffer = BytesIO()