🐛 small bugfixes

nateraw · Oct 10, 2022 · fe96f54 · fe96f54
1 parent 059036d
commit fe96f54
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/stable_diffusion_videos/__init__.py b/stable_diffusion_videos/__init__.py
@@ -105,7 +105,7 @@ def __dir__():
         "image_generation": [
             "generate_images",
             "upload_folder_chunked",
-        ]
+        ],
         "stable_diffusion_pipeline": [
             "StableDiffusionWalkPipeline",
             "NoCheck",

diff --git a/stable_diffusion_videos/image_generation.py b/stable_diffusion_videos/image_generation.py
@@ -168,7 +168,7 @@ def generate_images(
         print(f"Generating batch {batch_idx}")
 
         with torch.autocast('cuda'):
-            outputs = pipeline.step(
+            outputs = pipeline(
                 text_embeddings=embeds,
                 latents=noise,
                 num_inference_steps=num_inference_steps,