don't forget the vae scaling factor

anotherjesse · Nov 6, 2023 · 9d00e19 · 9d00e19
1 parent 2ff1ce5
commit 9d00e19
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -124,7 +124,9 @@ def predict(
             if consistency_decoder:
                 print("Running consistency decoder...")
                 start = time.time()
-                sample = self.consistency_decoder(sample.unsqueeze(0))
+                sample = self.consistency_decoder(
+                    sample.unsqueeze(0) / self.pipe.vae.config.scaling_factor
+                )
                 print("Consistency decoder took", time.time() - start, "seconds")
                 save_image(sample, output_path)
             else: