Upload app.py

app.py

@@ -75,6 +75,18 @@ DETAIL_LORA_FILE = os.environ.get(
 )
 DETAIL_LORA_SCALE = float(os.environ.get("H3_DETAIL_LORA_SCALE", "0.7"))
 
+# Spatial & Physics: object collisions, gravity, stacking, falling, occlusion, and
+# more consistent spatial relationships. The pruned H3 adapter does not require a trigger word.
+SPATIAL_LORA_REPO = os.environ.get(
+    "H3_SPATIAL_LORA_REPO",
+    "Jojocodex/minimax-h3-spatial-physics-lora",
+)
+SPATIAL_LORA_FILE = os.environ.get(
+    "H3_SPATIAL_LORA_FILE",
+    "wushu_spatial_physics_clean_3000_pruned.safetensors",
+)
+SPATIAL_LORA_SCALE = float(os.environ.get("H3_SPATIAL_LORA_SCALE", "0.4"))
+
 BLOWJOB_LORA_REPO = os.environ.get(
     "H3_BLOWJOB_LORA_REPO",
     "iamgroot1212/minimax-h3-loras",
@@ -579,6 +591,13 @@ def load_and_apply_extra_loras(transformer) -> list[str]:
             "Perfect Eyes / Skin / Hands LoRA",
             DETAIL_LORA_SCALE,
         ),
+        _apply_extra_lora(
+            transformer,
+            SPATIAL_LORA_REPO,
+            SPATIAL_LORA_FILE,
+            "Spatial & Physics LoRA",
+            SPATIAL_LORA_SCALE,
+        ),
         _apply_extra_lora(
             transformer,
             BLOWJOB_LORA_REPO,
@@ -620,8 +639,8 @@ def load_models() -> str | None:
         lora_status = load_and_apply_lora(pipe.transformer)
         print(f"[gen] {lora_status}", flush=True)
 
-        # Then fold the requested Motion Enhancer and BlowJob LoRA into the
-        # same transformer. They add their deltas on top of FlashGen.
+        # Then fold the requested extra LoRAs into the same transformer.
+        # They add their deltas on top of FlashGen.
         for extra_status in load_and_apply_extra_loras(pipe.transformer):
             print(f"[gen] {extra_status}", flush=True)