🐛 couple quick bugfixes

nateraw · Oct 10, 2022 · d4f2258 · d4f2258
1 parent b7e176d
commit d4f2258
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -105,6 +105,7 @@ video_path = pipeline.walk(
     num_interpolation_steps=num_interpolation_steps,
     audio_filepath='audio.mp3',
     audio_start_sec=audio_offsets[0],
+    fps=fps,
     height=512,  # use multiples of 64 if > 512. Multiples of 8 if < 512.
     width=512,   # use multiples of 64 if > 512. Multiples of 8 if < 512.
     output_dir='dreams',        # Where images/videos will be saved

diff --git a/stable_diffusion_videos/image_generation.py b/stable_diffusion_videos/image_generation.py
@@ -176,6 +176,7 @@ def generate_images(
                 eta=eta,
                 height=height,
                 width=width,
+                output_type="pil" if not upsample else "numpy",
             )['sample']
             if upsample:
                 images = []