Fixes

pipeline.py

@@ -27,12 +27,12 @@ class PreTrainedPipeline():
             raise ValueError("Input is not in ImageNet")
 
 
-        noise_vector = truncated_noise_sample(truncation=truncation, batch_size=1)
+        noise_vector = truncated_noise_sample(truncation=self.truncation, batch_size=1)
 
         noise_vector = torch.from_numpy(noise_vector)
         class_vector = torch.from_numpy(class_vector)
 
         with torch.no_grad():
-            output = model(noise_vector, class_vector, truncation)
+            output = self.model(noise_vector, class_vector, truncation)
 
         return transforms.ToPILImage()(output[0])