From 71e2e68c8e8437528c4488c9513fed12a93b6122 Mon Sep 17 00:00:00 2001
From: DeepBeepMeep <deepbeepmeep@yahoo.com>
Date: Tue, 11 Mar 2025 08:23:15 +0100
Subject: [PATCH] fixed bug with i2v introduced in last release

---
 gradio_server.py | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/gradio_server.py b/gradio_server.py
index 662ea50..fa55cd4 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -57,12 +57,12 @@ def _parse_args():
     )
 
 
-    # parser.add_argument(
-    #     "--lora-dir-i2v",
-    #     type=str,
-    #     default="loras_i2v",
-    #     help="Path to a directory that contains Loras for i2v"
-    # )
+    parser.add_argument(
+        "--lora-dir-i2v",
+        type=str,
+        default="loras_i2v",
+        help="Path to a directory that contains Loras for i2v"
+    )
 
 
     parser.add_argument(
@@ -236,11 +236,9 @@ if args.t2v:
 if args.i2v:
     use_image2video = True
 
-# if use_image2video:
-#     lora_dir =args.lora_dir_i2v
-#     lora_preselected_preset = args.lora_preset_i2v
-# else:
 lora_dir =args.lora_dir
+if use_image2video and len(lora_dir)==0:
+    lora_dir =args.lora_dir_i2v
 if len(lora_dir) ==0:
     lora_dir = "loras_i2v" if use_image2video else "loras"
 lora_preselected_preset = args.lora_preset
@@ -1248,7 +1246,7 @@ def create_demo():
                             label="Images as a starting point for new videos", type ="pil", #file_types= "image", 
                             columns=[3], rows=[1], object_fit="contain", height="auto", selected_index=0, interactive= True, visible=use_image2video)
                 else:
-                    image_to_continue = gr.Image(label= "Image as a starting point for a new video", visible=use_image2video)
+                    image_to_continue = gr.Image(label= "Image as a starting point for a new video", type ="pil", visible=use_image2video)
 
                 prompt = gr.Textbox(label="Prompts (multiple prompts separated by carriage returns will generate multiple videos, lines that starts with # are ignored)", value=default_prompt, lines=3)