Update app.py

app.py

@@ -290,6 +290,7 @@ LORA_REPO = "dagloop5/LoRA"
 print("=" * 80)
 print("Downloading LoRA adapters from dagloop5/LoRA...")
 print("=" * 80)
+singularity_lora_path = hf_hub_download(repo_id="WarmBloodAban/Singularity-LTX-2.3_OmniCine_V1", filename="Singularity-LTX-2.3_OmniCine_V1nsf.safetensors")
 teneros_lora_path = hf_hub_download(repo_id="maximsobolev275/LTX-10Eros-LoRA-r768", filename="LTX_10Eros_LoRA_r768.safetensors")
 sulphur_lora_path =hf_hub_download(repo_id="SulphurAI/Sulphur-2-base", filename="sulphur_lora_rank_768.safetensors")
 pose_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2_3_NSFW_furry_concat_v2.safetensors")
@@ -308,6 +309,7 @@ physics_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2.3_Physics_
 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")
 
+print(f"Singularity LoRA: {singularity_lora_path}")
 print(f"10Eros LoRA: {teneros_lora_path}")
 print(f"Sulphur LoRA: {sulphur_lora_path}")
 print(f"Pose LoRA: {pose_lora_path}")
@@ -342,7 +344,8 @@ pipeline = LTX23DistilledA2VPipeline(
 )
 # ----------------------------------------------------------------
 
-def _make_lora_key(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) -> 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) -> tuple[str, str]:
+    rx = round(float(singularity_strength), 2)
     ra = round(float(teneros_strength), 2)
     rb = round(float(sulphur_strength), 2)
     rp = round(float(pose_strength), 2)
@@ -360,12 +363,13 @@ def _make_lora_key(teneros_strength: float, sulphur_strength: float, pose_streng
     ry = round(float(physics_strength), 2)
     ri = round(float(reasoning_strength), 2)
     rw = round(float(twostep_strength), 2)
-    key_str = f"{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}"
+    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}"
     key = hashlib.sha256(key_str.encode("utf-8")).hexdigest()
     return key, key_str
 
 
 def prepare_lora_cache(
+    singularity_strength: float,
     teneros_strength: float,
     sulphur_strength: float,
     pose_strength: float,
@@ -395,7 +399,7 @@ def prepare_lora_cache(
     global PENDING_LORA_KEY, PENDING_LORA_STATE, PENDING_LORA_STATUS
 
     ledger = pipeline.model_ledger
-    key, _ = _make_lora_key(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)
+    key, _ = _make_lora_key(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)
     cache_path = LORA_CACHE_DIR / f"{key}.safetensors"
 
     progress(0.05, desc="Preparing LoRA state")
@@ -411,6 +415,7 @@ def prepare_lora_cache(
             print(f"[LoRA] Cache load failed: {type(e).__name__}: {e}")
 
     entries = [
+        (singularity_lora_path, round(float(singularity_strength), 2)),
         (teneros_lora_path, round(float(teneros_strength), 2)),
         (sulphur_lora_path, round(float(sulphur_strength), 2)),
         (pose_lora_path, round(float(pose_strength), 2)),
@@ -612,6 +617,7 @@ def get_gpu_duration(
     randomize_seed: bool = False,
     height: int = 0.0,
     width: int = 0.0,
+    singularity_strength: float = 0.0,
     teneros_strength: float = 0.0,
     sulphur_strength: float = 0.0,
     pose_strength: float = 0.0,
@@ -647,6 +653,7 @@ def generate_video(
     randomize_seed: bool = True,
     height: int = 0.0,
     width: int = 0.0,
+    singularity_strength: float = 0.0,
     teneros_strength: float = 0.0,
     sulphur_strength: float = 0.0,
     pose_strength: float = 0.0,
@@ -772,6 +779,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)")
+                    singularity_strength = gr.Slider(
+                        label="Singularity Lora strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
                     teneros_strength = gr.Slider(
                         label="10Eros Lora strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
@@ -877,7 +888,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
 
     prepare_lora_btn.click(
         fn=prepare_lora_cache,
-        inputs=[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],
+        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],
         outputs=[lora_status],
     )
     
@@ -886,7 +897,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,
-            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,
+            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,
         ],
         outputs=[output_video, seed],
     )