LORA

app.py

@@ -123,6 +123,7 @@ input_image_debug_value = [None]
 input_video_debug_value = [None]
 prompt_debug_value = [None]
 total_second_length_debug_value = [None]
+lora_file_debug_value = [None]
 
 default_local_storage = {
         "generation-mode": "image",
@@ -1016,12 +1017,13 @@ def process(input_image,
     if auto_allocation:
         allocation_time = min(total_second_length * 60 * (1.5 if use_teacache else 3.0) * (1 + ((steps - 25) / 25))**2, 600)
 
-    if input_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
+    if input_image_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None or lora_file_debug_value[0] is not None:
         input_image = input_image_debug_value[0]
         prompt = prompt_debug_value[0]
         total_second_length = total_second_length_debug_value[0]
         allocation_time = min(total_second_length_debug_value[0] * 60 * 100, 600)
-        input_image_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = None
+        lora_file = lora_file_debug_value[0]
+        input_image_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = lora_file_debug_value[0] = None
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
@@ -1133,12 +1135,13 @@ def process_video(input_video, prompt, n_prompt, randomize_seed, seed, auto_allo
     if auto_allocation:
         allocation_time = min(total_second_length * 60 * (2.5 if use_teacache else 3.5) * (1 + ((steps - 25) / 25))**2, 600)
 
-    if input_video_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None:
+    if input_video_debug_value[0] is not None or prompt_debug_value[0] is not None or total_second_length_debug_value[0] is not None or lora_file_debug_value is not None:
         input_video = input_video_debug_value[0]
         prompt = prompt_debug_value[0]
         total_second_length = total_second_length_debug_value[0]
         allocation_time = min(total_second_length_debug_value[0] * 60 * 100, 600)
-        input_video_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = None
+        lora_file = lora_file_debug_value[0]
+        input_video_debug_value[0] = prompt_debug_value[0] = total_second_length_debug_value[0] = lora_file_debug_value[0] = None
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
@@ -1345,6 +1348,7 @@ with block:
                 input_video_debug = gr.Video(sources='upload', label="Input Video Debug", height=320)
                 prompt_debug = gr.Textbox(label="Prompt Debug", value='')
                 total_second_length_debug = gr.Slider(label="Additional Video Length to Generate (seconds) Debug", minimum=1, maximum=120, value=1, step=0.1)
+                lora_file_debug = gr.File(label="LoRA File", file_count="single", type="filepath")
 
         with gr.Column():
             warning = gr.HTML(elem_id="warning", value = "<center><big>Your computer must <u>not</u> enter into standby mode.</big><br/>On Chrome, you can force to keep a tab alive in <code>chrome://discards/</code></center>", visible = False)
@@ -1848,35 +1852,42 @@ with block:
             return [gr.update(visible = False), gr.update(visible = False), gr.update(visible = False), gr.update(visible = True), gr.update(visible = False), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = True), gr.update(visible = False)]
 
     
-    def handle_field_debug_change(input_image_debug_data, input_video_debug_data, prompt_debug_data, total_second_length_debug_data):
+    def handle_field_debug_change(input_image_debug_data, input_video_debug_data, prompt_debug_data, total_second_length_debug_data, lora_file_debug_data):
         print("handle_field_debug_change")
         input_image_debug_value[0] = input_image_debug_data
         input_video_debug_value[0] = input_video_debug_data
         prompt_debug_value[0] = prompt_debug_data
         total_second_length_debug_value[0] = total_second_length_debug_data
+        lora_file_debug_value[0] = lora_file_debug_data
         return []
 
     input_image_debug.upload(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug, lora_file_debug],
         outputs=[]
     )
     
     input_video_debug.upload(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug, lora_file_debug],
         outputs=[]
     )
     
     prompt_debug.change(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug, lora_file_debug],
         outputs=[]
     )
     
     total_second_length_debug.change(
         fn=handle_field_debug_change,
-        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug],
+        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug, lora_file_debug],
+        outputs=[]
+    )
+
+    lora_file_debug.upload(
+        fn=handle_field_debug_change,
+        inputs=[input_image_debug, input_video_debug, prompt_debug, total_second_length_debug, lora_file_debug],
         outputs=[]
     )