diff --git a/README.md b/README.md
index 33a231f..3ff90ce 100644
--- 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
index c212cb1..82cd7da 100644
--- 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 = []