Update app.py

app.py

@@ -302,13 +302,18 @@ dramatic_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX-2.3 - Orga
 fluid_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="Cr3ampi3_animation_sulphur-2_i2v_v1.0.safetensors") # cr3ampi3 animation
 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")
+voice_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="hentai_voice_ltx23_v2.comfy.safetensors")
 realism_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="FurryenhancerLTX2.3V4.094fused.safetensors")
 transition_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX-2_takerpov_lora_v1.2.safetensors") # takerpov1, taker pov
 physics_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2.3_Physics_V2_000002000.safetensors")
 reasoning_lora_path = hf_hub_download(repo_id="LiconStudio/Ltx2.3-VBVR-lora-I2V", filename="Ltx2.3-Licon-VBVR-I2V-390K-R32.safetensors")
 twostep_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2.3_Multi_step_video_reasoning_V0.1.safetensors")
 mcfurry_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX-2_3_mc_Furry_realistic_lora_v1.safetensors")
+dm_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="Doggy_mission_sulphur-2_v0.5.safetensors")
+praxis_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="Penile_Praxis_V4.safetensors")
+threed_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="ltx2-3d-animations-12500-steps-k3nk.safetensors")
+concept_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="ltx23_nsfw_helper_multi_concept_lora_v2.safetensors")
+bulge_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="stomach_bulge_10eros_sulphur_v1.safetensors")
 
 print(f"Singularity LoRA: {singularity_lora_path}")
 print(f"10Eros LoRA: {teneros_lora_path}")
@@ -329,6 +334,11 @@ print(f"Physics LoRA: {physics_lora_path}")
 print(f"Reasoning LoRA: {reasoning_lora_path}")
 print(f"Twostep LoRA: {twostep_lora_path}")
 print(f"Mcfurry LoRA: {mcfurry_lora_path}")
+print(f"DM LoRA: {dm_lora_path}")
+print(f"Praxis LoRA: {praxis_lora_path}")
+print(f"ThreeD LoRA: {threed_lora_path}")
+print(f"Concept LoRA: {concept_lora_path}")
+print(f"Bulge LoRA: {bulge_lora_path}")
 # ----------------------------------------------------------------
 
 print(f"Checkpoint: {checkpoint_path}")
@@ -346,7 +356,7 @@ pipeline = LTX23DistilledA2VPipeline(
 )
 # ----------------------------------------------------------------
 
-def _make_lora_key(singularity_strength: float, teneros_strength: float, sulphur_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, physics_strength: float, reasoning_strength: float, twostep_strength: float, mcfurry_strength: float) -> tuple[str, str]:
+def _make_lora_key(singularity_strength: float, teneros_strength: float, sulphur_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, physics_strength: float, reasoning_strength: float, twostep_strength: float, mcfurry_strength: float, dm_strength: float, praxis_strength: float, threed_strength: float, concept_strength: float, bulge_strength: float) -> tuple[str, str]:
     rx = round(float(singularity_strength), 2)
     ra = round(float(teneros_strength), 2)
     rb = round(float(sulphur_strength), 2)
@@ -366,7 +376,12 @@ def _make_lora_key(singularity_strength: float, teneros_strength: float, sulphur
     ri = round(float(reasoning_strength), 2)
     rw = round(float(twostep_strength), 2)
     mc = round(float(mcfurry_strength), 2)
-    key_str = f"{singularity_lora_path}:{rx}|{teneros_lora_path}:{ra}|{sulphur_lora_path}:{rb}|{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}|{physics_lora_path}:{ry}|{reasoning_lora_path}:{ri}|{twostep_lora_path}:{rw}|{mcfurry_lora_path}:{mc}"
+    dm = round(float(dm_strength), 2)
+    pr = round(float(praxis_strength), 2)
+    td = round(float(threed_strength), 2)
+    co = round(float(concept_strength), 2)
+    bu = round(float(bulge_strength), 2)
+    key_str = f"{singularity_lora_path}:{rx}|{teneros_lora_path}:{ra}|{sulphur_lora_path}:{rb}|{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}|{physics_lora_path}:{ry}|{reasoning_lora_path}:{ri}|{twostep_lora_path}:{rw}|{mcfurry_lora_path}:{mc}|{dm_lora_path}:{dm}|{praxis_lora_path}:{pr}|{threed_lora_path}:{td}|{concept_lora_path}:{co}|{bulge_lora_path}:{bu}"
     key = hashlib.sha256(key_str.encode("utf-8")).hexdigest()
     return key, key_str
 
@@ -391,6 +406,11 @@ def prepare_lora_cache(
     reasoning_strength: float,
     twostep_strength: float,
     mcfurry_strength: float,
+    dm_strength: float,
+    praxis_strength: float,
+    threed_strength: float,
+    concept_strength: float,
+    bulge_strength: float,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -438,6 +458,11 @@ def prepare_lora_cache(
         (reasoning_lora_path, round(float(reasoning_strength), 2)),
         (twostep_lora_path, round(float(twostep_strength), 2)),
         (mcfurry_lora_path, round(float(mcfurry_strength), 2)),
+        (dm_lora_path, round(float(dm_strength), 2)),
+        (praxis_lora_path, round(float(praxis_strength), 2)),
+        (threed_lora_path, round(float(threed_strength), 2)),
+        (concept_lora_path, round(float(concept_strength), 2)),
+        (bulge_lora_path, round(float(bulge_strength), 2)),
     ]
     loras_for_builder = [
         LoraPathStrengthAndSDOps(path, strength, LTXV_LORA_COMFY_RENAMING_MAP)
@@ -641,6 +666,11 @@ def get_gpu_duration(
     reasoning_strength: float = 0.0,
     twostep_strength: float = 0.0,
     mcfurry_strength: float = 0.0,
+    dm_strength: float = 0.0,
+    praxis_strength: float = 0.0,
+    threed_strength: float = 0.0,
+    concept_strength: float = 0.0,
+    bulge_strength: float = 0.0,
     progress=None, 
 ):
     return int(gpu_duration)
@@ -678,6 +708,11 @@ def generate_video(
     reasoning_strength: float = 0.0,
     twostep_strength: float = 0.0,
     mcfurry_strength: float = 0.0,
+    dm_strength: float = 0.0,
+    praxis_strength: float = 0.0,
+    threed_strength: float = 0.0,
+    concept_strength: float = 0.0,
+    bulge_strength: float = 0.0,
     progress=gr.Progress(track_tqdm=True),
 ):
     try:
@@ -862,6 +897,26 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                         label="t2v anthro realism strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
                     )
+                    dm_strength = gr.Slider(
+                        label="DM3D strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    praxis_strength = gr.Slider(
+                        label="Praxis strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    threed_strength = gr.Slider(
+                        label="3D animation strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    concept_strength = gr.Slider(
+                        label="Conceptual strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    bulge_strength = gr.Slider(
+                        label="Bulge strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
                 prepare_lora_btn = gr.Button("Prepare / Load LoRA Cache", variant="secondary")
                 lora_status = gr.Textbox(
                     label="LoRA Cache Status",
@@ -899,7 +954,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
 
     prepare_lora_btn.click(
         fn=prepare_lora_cache,
-        inputs=[singularity_strength, teneros_strength, sulphur_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, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength],
+        inputs=[singularity_strength, teneros_strength, sulphur_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, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength, dm_strength, praxis_strength, threed_strength, concept_strength, bulge_strength],
         outputs=[lora_status],
     )
     
@@ -908,7 +963,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,
-            singularity_strength, teneros_strength, sulphur_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, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength,
+            singularity_strength, teneros_strength, sulphur_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, physics_strength, reasoning_strength, twostep_strength, mcfurry_strength, dm_strength, praxis_strength, threed_strength, concept_strength, bulge_strength,
         ],
         outputs=[output_video, seed],
     )