Update app.py

app.py

@@ -299,7 +299,7 @@ print("Downloading LoRA adapters from dagloop5/LoRA...")
 print("=" * 80)
 paste_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="ltx23-kabapaste.safetensors")
 ruri_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX23-ruri.safetensors")
-transit_lora_path = hf_hub_download(repo_id="valiantcat/LTX-2.3-Transition-LORA", filename="ltx2.3-transition.safetensors")
+popping_lora_path = hf_hub_download(repo_id="YuLexuan30/Balloon_popping", filename="poppedv2ltx23.safetensors")
 pose_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2_3_NSFW_furry_concat_v2.safetensors")
 general_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2.3_Reasoning_V1.safetensors")
 motion_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="motion_helper.safetensors")
@@ -313,7 +313,7 @@ sulphur_lora_path = hf_hub_download(repo_id="SulphurAI/Sulphur-2-base", filename
 
 print(f"Paste LoRA: {paste_lora_path }")
 print(f"Ruri LoRA: {ruri_lora_path }")
-print(f"Transit LoRA: {transit_lora_path }")
+print(f"Popping LoRA: {popping_lora_path }")
 print(f"Pose LoRA: {pose_lora_path}")
 print(f"General LoRA: {general_lora_path}")
 print(f"Motion LoRA: {motion_lora_path}")
@@ -341,10 +341,10 @@ pipeline = LTX23DistilledA2VPipeline(
 )
 # ----------------------------------------------------------------
 
-def _make_lora_key(paste_strength: float,ruri_strength: float,transit_strength: float,pose_strength: float, general_strength: float, motion_strength: float, dreamlay_strength: float, mself_strength: float, dramatic_strength: float, fluid_strength: float, liquid_strength: float, demopose_strength: float, sulphur_strength: float) -> tuple[str, str]:
+def _make_lora_key(paste_strength: float,ruri_strength: float,popping_strength: float,pose_strength: float, general_strength: float, motion_strength: float, dreamlay_strength: float, mself_strength: float, dramatic_strength: float, fluid_strength: float, liquid_strength: float, demopose_strength: float, sulphur_strength: float) -> tuple[str, str]:
     ps = round(float(paste_strength), 2)
     rr = round(float(ruri_strength), 2)
-    rt = round(float(transit_strength), 2)
+    rt = round(float(popping_strength), 2)
     rp = round(float(pose_strength), 2)
     rg = round(float(general_strength), 2)
     rm = round(float(motion_strength), 2)
@@ -355,7 +355,7 @@ def _make_lora_key(paste_strength: float,ruri_strength: float,transit_strength:
     rl = round(float(liquid_strength), 2)
     ro = round(float(demopose_strength), 2)
     ss = round(float(sulphur_strength), 2)
-    key_str = f"{paste_lora_path}:{ps}|{ruri_lora_path}:{rr}|{transit_lora_path}:{rt}|{pose_lora_path}:{rp}|{general_lora_path}:{rg}|{motion_lora_path}:{rm}|{dreamlay_lora_path}:{rd}|{mself_lora_path}:{rs}|{dramatic_lora_path}:{rr}|{fluid_lora_path}:{rf}|{liquid_lora_path}:{rl}|{demopose_lora_path}:{ro}|{sulphur_lora_path}:{ss}"
+    key_str = f"{paste_lora_path}:{ps}|{ruri_lora_path}:{rr}|{popping_lora_path}:{rt}|{pose_lora_path}:{rp}|{general_lora_path}:{rg}|{motion_lora_path}:{rm}|{dreamlay_lora_path}:{rd}|{mself_lora_path}:{rs}|{dramatic_lora_path}:{rr}|{fluid_lora_path}:{rf}|{liquid_lora_path}:{rl}|{demopose_lora_path}:{ro}|{sulphur_lora_path}:{ss}"
     key = hashlib.sha256(key_str.encode("utf-8")).hexdigest()
     return key, key_str
 
@@ -363,7 +363,7 @@ def _make_lora_key(paste_strength: float,ruri_strength: float,transit_strength:
 def prepare_lora_cache(
     paste_strength: float,
     ruri_strength: float,
-    transit_strength: float,
+    popping_strength: float,
     pose_strength: float,
     general_strength: float,
     motion_strength: float,
@@ -386,7 +386,7 @@ def prepare_lora_cache(
     global PENDING_LORA_KEY, PENDING_LORA_STATE, PENDING_LORA_STATUS
 
     ledger = pipeline.model_ledger
-    key, _ = _make_lora_key(paste_strength,ruri_strength,transit_strength,pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength)
+    key, _ = _make_lora_key(paste_strength,ruri_strength,popping_strength,pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength)
     cache_path = LORA_CACHE_DIR / f"{key}.pt"
 
     progress(0.05, desc="Preparing LoRA state")
@@ -404,7 +404,7 @@ def prepare_lora_cache(
     entries = [
         (paste_lora_path, round(float(paste_strength), 2)),
         (ruri_lora_path, round(float(ruri_strength), 2)),
-        (transit_lora_path, round(float(transit_strength), 2)),
+        (popping_lora_path, round(float(popping_strength), 2)),
         (pose_lora_path, round(float(pose_strength), 2)),
         (general_lora_path, round(float(general_strength), 2)),
         (motion_lora_path, round(float(motion_strength), 2)),
@@ -603,7 +603,7 @@ def get_gpu_duration(
     width: int = 1536,
     paste_strength: float = 0.0,
     ruri_strength: float = 0.0,
-    transit_strength: float = 0.0,
+    popping_strength: float = 0.0,
     pose_strength: float = 0.0,
     general_strength: float = 0.0,
     motion_strength: float = 0.0,
@@ -634,7 +634,7 @@ def generate_video(
     width: int = 1536,
     paste_strength: float = 0.0,
     ruri_strength: float = 0.0,
-    transit_strength: float = 0.0,
+    popping_strength: float = 0.0,
     pose_strength: float = 0.0,
     general_strength: float = 0.0,
     motion_strength: float = 0.0,
@@ -753,7 +753,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                     gr.Markdown("### LoRA adapter strengths (set to 0 to disable)")
                     paste_strength = gr.Slider(label="Paste anything Lora strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
                     ruri_strength = gr.Slider(label="Ruri lady Lora strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
-                    transit_strength = gr.Slider(label="Transition Lora strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
+                    popping_strength = gr.Slider(label="Balloon popping Lora strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
                     pose_strength = gr.Slider(label="Anthro Enhancer strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
                     general_strength = gr.Slider(label="Reasoning Enhancer strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
                     motion_strength = gr.Slider(label="Anthro Posing Helper strength", minimum=0.0, maximum=2.0, value=0.0, step=0.01)
@@ -851,7 +851,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
 
     prepare_lora_btn.click(
         fn=prepare_lora_cache,
-        inputs=[paste_strength, ruri_strength, transit_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength],
+        inputs=[paste_strength, ruri_strength, popping_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength],
         outputs=[lora_status],
     )
     
@@ -860,7 +860,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
         inputs=[
             first_image, last_image, input_audio, prompt, duration, gpu_duration, enhance_prompt,
             seed, randomize_seed, height, width,
-            paste_strength, ruri_strength, transit_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength,
+            paste_strength, ruri_strength, popping_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, sulphur_strength,
         ],
         outputs=[output_video, seed],
     )