Update app.py

app.py

@@ -338,6 +338,7 @@ LORA_REPO = "dagloop5/LoRA"
 print("=" * 80)
 print("Downloading LoRA adapters from dagloop5/LoRA...")
 print("=" * 80)
+distilled_lora_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename=DISTILLED_LORA_FILE)
 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_VBVR_Reasoning_I2V_V2.safetensors")
 motion_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="motion_helper.safetensors")
@@ -345,9 +346,13 @@ dreamlay_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="DR34ML4Y_LTXXX
 mself_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="Furry Hyper Masturbation - LTX-2 I2V v1.safetensors") # Hyperfap
 dramatic_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX-2.3 - Orgasm.safetensors") # "[He | She] is having am orgasm." (am or an?)
 fluid_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="cr3ampi3_animation_i2v_ltx2_v1.0.safetensors") # cr3ampi3 animation., missionary animation, doggystyle bouncy animation, double penetration animation
-liquid_lora_path = hf_hub_download(repo_id="valiantcat/LTX-2.3-Transition-LORA", filename="ltx2.3-transition.safetensors") # wet dr1pp
-demopose_lora_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename=DISTILLED_LORA_FILE)
+liquid_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="liquid_wet_dr1pp_ltx2_v1.0_scaled.safetensors") # wet dr1pp
+demopose_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="clapping-cheeks-audio-v001-alpha.safetensors")
+voice_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="hentai_voice_ltx23.safetensors")
+realism_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="FurryenhancerLTX2.3V1.215.safetensors")
+transition_lora_path = hf_hub_download(repo_id="valiantcat/LTX-2.3-Transition-LORA", filename="ltx2.3-transition.safetensors")
 
+print(f"Distilled LoRA: {distilled_lora_path}")
 print(f"Pose LoRA: {pose_lora_path}")
 print(f"General LoRA: {general_lora_path}")
 print(f"Motion LoRA: {motion_lora_path}")
@@ -357,6 +362,9 @@ print(f"Dramatic LoRA: {dramatic_lora_path}")
 print(f"Fluid LoRA: {fluid_lora_path}")
 print(f"Liquid LoRA: {liquid_lora_path}")
 print(f"Demopose LoRA: {demopose_lora_path}")
+print(f"Voice LoRA: {voice_lora_path}")
+print(f"Realism LoRA: {realism_lora_path}")
+print(f"Transition LoRA: {transition_lora_path}")
 # ----------------------------------------------------------------
 
 print(f"Checkpoint: {checkpoint_path}")
@@ -374,7 +382,8 @@ pipeline = LTX23DistilledA2VPipeline(
 )
 # ----------------------------------------------------------------
 
-def _make_lora_key(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) -> tuple[str, str]:
+def _make_lora_key(distilled_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, voice_strength: float, realism_strength: float, transition_strength: float) -> tuple[str, str]:
+    rn = round(float(pose_strength), 2)
     rp = round(float(pose_strength), 2)
     rg = round(float(general_strength), 2)
     rm = round(float(motion_strength), 2)
@@ -384,12 +393,16 @@ def _make_lora_key(pose_strength: float, general_strength: float, motion_strengt
     rf = round(float(fluid_strength), 2)
     rl = round(float(liquid_strength), 2)
     ro = round(float(demopose_strength), 2)
-    key_str = f"{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}"
+    rv = round(float(voice_strength), 2)
+    re = round(float(realism_strength), 2)
+    rt = round(float(transition_strength), 2)
+    key_str = f"{distilled_lora_path}:{rn}|{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}|{voice_lora_path}:{rv}|{realism_lora_path}:{re}|{transition_lora_path}:{rt}"
     key = hashlib.sha256(key_str.encode("utf-8")).hexdigest()
     return key, key_str
 
 
 def prepare_lora_cache(
+    distilled_strength: float,
     pose_strength: float,
     general_strength: float,
     motion_strength: float,
@@ -399,6 +412,9 @@ def prepare_lora_cache(
     fluid_strength: float,
     liquid_strength: float,
     demopose_strength: float,
+    voice_strength: float,
+    realism_strength: float,
+    transition_strength: float,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -411,7 +427,7 @@ def prepare_lora_cache(
     global PENDING_LORA_KEY, PENDING_LORA_STATE, PENDING_LORA_STATUS
 
     ledger = pipeline.model_ledger
-    key, _ = _make_lora_key(pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength)
+    key, _ = _make_lora_key(distilled_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength)
     cache_path = LORA_CACHE_DIR / f"{key}.safetensors"
 
     progress(0.05, desc="Preparing LoRA state")
@@ -427,6 +443,7 @@ def prepare_lora_cache(
             print(f"[LoRA] Cache load failed: {type(e).__name__}: {e}")
 
     entries = [
+        (distilled_lora_path, round(float(distilled_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)),
@@ -436,6 +453,9 @@ def prepare_lora_cache(
         (fluid_lora_path, round(float(fluid_strength), 2)),
         (liquid_lora_path, round(float(liquid_strength), 2)),
         (demopose_lora_path, round(float(demopose_strength), 2)),
+        (voice_lora_path, round(float(voice_strength), 2)),
+        (realism_lora_path, round(float(realism_strength), 2)),
+        (transition_lora_path, round(float(transition_strength), 2)),
     ]
     loras_for_builder = [
         LoraPathStrengthAndSDOps(path, strength, LTXV_LORA_COMFY_RENAMING_MAP)
@@ -620,6 +640,7 @@ def get_gpu_duration(
     randomize_seed: bool = True,
     height: int = 1024,
     width: int = 1536,
+    distilled_strength: float = 0.0,
     pose_strength: float = 0.0,
     general_strength: float = 0.0,
     motion_strength: float = 0.0,
@@ -629,6 +650,9 @@ def get_gpu_duration(
     fluid_strength: float = 0.0,
     liquid_strength: float = 0.0,
     demopose_strength: float = 0.0,
+    voice_strength: float = 0.0,
+    realism_strength: float = 0.0,
+    transition_strength: float = 0.0,
     progress=None,
 ):
     return int(gpu_duration)
@@ -647,6 +671,7 @@ def generate_video(
     randomize_seed: bool = True,
     height: int = 1024,
     width: int = 1536,
+    distilled_strength: float = 0.0,
     pose_strength: float = 0.0,
     general_strength: float = 0.0,
     motion_strength: float = 0.0,
@@ -656,6 +681,9 @@ def generate_video(
     fluid_strength: float = 0.0,
     liquid_strength: float = 0.0,
     demopose_strength: float = 0.0,
+    voice_strength: float = 0.0,
+    realism_strength: float = 0.0,
+    transition_strength: float = 0.0,
     progress=gr.Progress(track_tqdm=True),
 ):
     try:
@@ -764,6 +792,10 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                     high_res = gr.Checkbox(label="High Resolution", value=True)
                 with gr.Column():
                     gr.Markdown("### LoRA adapter strengths (set to 0 to disable; slow and WIP)")
+                    distilled_strength = gr.Slider(
+                        label="Distilled LoRA strength (required!)",
+                        minimum=0.0, maximum=2.0, value=0.9, step=0.01
+                    )
                     pose_strength = gr.Slider(
                         label="Anthro Enhancer strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
@@ -793,11 +825,23 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
                     )
                     liquid_strength = gr.Slider(
-                        label="Transition Helper strength",
+                        label="Liquid Helper strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
                     )
                     demopose_strength = gr.Slider(
-                        label="Distilled strength",
+                        label="Audio Helper strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    voice_strength = gr.Slider(
+                        label="Voice Helper strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    realism_strength = gr.Slider(
+                        label="Anthro Realism strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    transition_strength = gr.Slider(
+                        label="Transition strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
                     )
                 prepare_lora_btn = gr.Button("Prepare / Load LoRA Cache", variant="secondary")
@@ -836,6 +880,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                 True,
                 1024,
                 1024,
+                0.0,
                 0.0,  # pose_strength (example)
                 0.0,  # general_strength (example)
                 0.0,  # motion_strength (example)
@@ -845,12 +890,15 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                 0.0,
                 0.0,
                 0.0,
+                0.0,
+                0.0,
+                0.0,
             ],
         ],
         inputs=[
             first_image, last_image, input_audio, prompt, duration, gpu_duration,
             enhance_prompt, seed, randomize_seed, height, width,
-            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength,
+            distilled_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength,        
         ],
     )
 
@@ -874,7 +922,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
 
     prepare_lora_btn.click(
         fn=prepare_lora_cache,
-        inputs=[pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength],
+        inputs=[distilled_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength],
         outputs=[lora_status],
     )
     
@@ -883,7 +931,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,
-            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength,
+            distilled_strength, pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength,
         ],
         outputs=[output_video, seed],
     )