🐛 fix upsample option in images interface

nateraw · Sep 12, 2022 · f93f06b · f93f06b
1 parent 159e9eb
commit f93f06b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/stable_diffusion_videos/app.py b/stable_diffusion_videos/app.py
@@ -28,6 +28,7 @@ def fn_images(
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
             generator=torch.Generator(device=pipeline.device).manual_seed(seed),
+            output_type='pil' if not upsample else 'numpy',
         )["sample"][0]
         return img if not upsample else upsampling_pipeline(img)