🐛 sample -> images

nateraw · Oct 20, 2022 · 6974165 · 6974165
1 parent 2870a42
commit 6974165
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable_diffusion_videos/image_generation.py b/stable_diffusion_videos/image_generation.py
@@ -182,7 +182,7 @@ def generate_images(
                 height=height,
                 width=width,
                 output_type="pil" if not upsample else "numpy",
-            )['sample']
+            )['images']
             if upsample:
                 images = []
                 for output in outputs: