Update app.py

app.py

@@ -37,6 +37,7 @@ GPU_SIZE = os.environ.get("H3_GPU_SIZE", "xlarge")
 
 LORA_REPO = os.environ.get("H3_LORA_REPO", "dagloop5/LoRA")
 LORA_FILES = {
+    "lora1": os.environ.get("H3_LORA_A_FILE", "minimax_h3_turbo_v4_step600_ema.safetensors"),
     "loraa": os.environ.get("H3_LORA_A_FILE", "Mylo_lora_epoch31.safetensors"),
     "lorab": os.environ.get("H3_LORA_B_FILE", "VBVR_H3_attn_only.safetensors"),
     "lorac": os.environ.get("H3_LORA_C_FILE", "AIO_V2.safetensors"),
@@ -48,6 +49,7 @@ LORA_FILES = {
 # Display names, keyed the same as LORA_FILES — used in the UI slider labels, the per-request report line, and
 # the status line's failure list. Keep these two dicts' keys in sync when adding a LoRA.
 LORA_LABELS = {
+    "lora1": "Distilled LoRA",
     "loraa": "Anthro Enhancer",
     "lorab": "Reasoning Enhancer",
     "lorac": "HM-AIO", # hmmotion
@@ -56,6 +58,7 @@ LORA_LABELS = {
     "loraf": "Moaxx", # moawxx
     "lorag": "Fluid Enhancer",
 }
+DEFAULT_LORA_1_STRENGTH = 0.0
 DEFAULT_LORA_A_STRENGTH = 0.0
 DEFAULT_LORA_B_STRENGTH = 0.0
 DEFAULT_LORA_C_STRENGTH = 0.0
@@ -542,6 +545,7 @@ def generate(
     interpolation=DEFAULT_INTERPOLATION,
     seed=42,
     upsample=False,
+    lora_1_strength=DEFAULT_LORA_1_STRENGTH,
     lora_a_strength=DEFAULT_LORA_A_STRENGTH,
     lora_b_strength=DEFAULT_LORA_B_STRENGTH,
     lora_c_strength=DEFAULT_LORA_C_STRENGTH,
@@ -590,7 +594,7 @@ def generate(
 
     # Every UI LoRA slider gets packed into one dict here — this is the only place a new LoRA's slider value
     # needs wiring in; `_generate`, `set_adapters`, and the report line below are all keyed off this dict.
-    lora_strengths = {"loraa": float(lora_a_strength), "lorab": float(lora_b_strength), "lorac": float(lora_c_strength), "lorad": float(lora_d_strength), "lorae": float(lora_e_strength), "loraf": float(lora_f_strength), "lorag": float(lora_g_strength)}
+    lora_strengths = {"lora1": float(lora_1_strength), "loraa": float(lora_a_strength), "lorab": float(lora_b_strength), "lorac": float(lora_c_strength), "lorad": float(lora_d_strength), "lorae": float(lora_e_strength), "loraf": float(lora_f_strength), "lorag": float(lora_g_strength)}
 
     progress(0.1, desc=f"Denoising {int(steps)} steps at {width}x{height}, {num_frames} frames ...")
     call = (
@@ -733,6 +737,13 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
                     value=DEFAULT_STEPS,
                     info="PlagueKind: 15-20 on the linear_quadratic grid.",
                 )
+                lora_1_strength = gr.Slider(
+                    label="Distilled LoRA",
+                    minimum=0.0,
+                    maximum=2.0,
+                    step=0.05,
+                    value=DEFAULT_LORA_1_STRENGTH,
+                )
                 lora_a_strength = gr.Slider(
                     label="Anthro Enhancer LoRA",
                     minimum=0.0,
@@ -828,6 +839,7 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         interpolation,
         seed,
         upsample,
+        lora_1_strength,
         lora_a_strength,
         lora_b_strength,
         lora_c_strength,