runner: Fixes to image-to-image/video for modal

livepeer · Feb 29, 2024 · 1d9b52b · 1d9b52b
1 parent 766ba4c
commit 1d9b52b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/runner/app/routes/image_to_image.py b/runner/app/routes/image_to_image.py
@@ -72,8 +72,8 @@ async def image_to_image(
 
     try:
         images = pipeline(
-            prompt,
-            image,
+            prompt=prompt,
+            image=image,
             strength=strength,
             guidance_scale=guidance_scale,
             negative_prompt=negative_prompt,

diff --git a/runner/app/routes/image_to_video.py b/runner/app/routes/image_to_video.py
@@ -60,7 +60,7 @@ async def image_to_video(
 
     try:
         batch_frames = pipeline(
-            Image.open(image.file).convert("RGB"),
+            image=Image.open(image.file).convert("RGB"),
             height=height,
             width=width,
             fps=fps,