Update app1.py

app1.py

@@ -58,16 +58,25 @@ pipe = WanPipeline.from_pretrained(
 # Notes:
 # - We attach this LoRA only to transformer_2 (the low-noise stage for Wan2.2).
 # - Adapter name `orbit_low` lets you adjust/enable later via set_adapters if needed.
-pipe.load_lora_weights(
-    "ostris/wan22_i2v_14b_orbit_shot_lora",
-    weight_name="wan22_14b_i2v_orbit_low_noise.safetensors",
-    adapter_name="orbit_low",
-    components=["transformer_2"],  # IMPORTANT: apply only to low-noise stage
-)
-# Activate the adapter on transformer_2 with weight 1.0 (changeable)
-pipe.set_adapters(
-    ["orbit_low"], adapter_weights=[1.0], components=["transformer_2"]
-)
+# pipe.load_lora_weights(
+#     "ostris/wan22_i2v_14b_orbit_shot_lora",
+#     weight_name="wan22_14b_i2v_orbit_low_noise.safetensors",
+#     adapter_name="orbit_low",
+#     components=["transformer_2"],  # IMPORTANT: apply only to low-noise stage
+# )
+# # Activate the adapter on transformer_2 with weight 1.0 (changeable)
+# pipe.set_adapters(
+#     ["orbit_low"], adapter_weights=[1.0], components=["transformer_2"]
+# )
+
+
+LORA_REPO_ID = "ostris/wan22_i2v_14b_orbit_shot_lora"
+LORA_FILENAME = "wan22_14b_i2v_orbit_low_noise.safetensors"
+
+
+
+
+    
 # -------------------------------------------------------------------------------
 
 # Usual CUDA cleanup
@@ -88,20 +97,18 @@ optimize_pipeline_(
 # ---- NEW: Load + Fuse Orbit-Shot LoRA on the LOW-NOISE stage only (transformer_2) ----
 # Repo: ostris/wan22_i2v_14b_orbit_shot_lora
 # File: wan22_14b_i2v_orbit_low_noise.safetensors
-
-pipe.load_lora_weights(
-    "ostris/wan22_i2v_14b_orbit_shot_lora",
-    weight_name="wan22_14b_i2v_orbit_low_noise.safetensors",
-    adapter_name="orbit_low",
-    components=["transformer_2"],  # apply only to low-noise stage
-)
-
-# Fuse permanently into transformer_2 at scale 1.0
-pipe.fuse_lora(
-    adapter_names=["orbit_low"],
-    lora_scale=1.0,
-    components=["transformer_2"],
-)
+try:
+    causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
+    print("✅ LoRA downloaded to:", causvid_path)
+
+    pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
+    pipe.set_adapters(["causvid_lora"], adapter_weights=[0.75])
+    pipe.fuse_lora()
+
+except Exception as e:
+    import traceback
+    print("❌ Error during LoRA loading:")
+    traceback.print_exc()
 # -------------------------------------------------------------------------------