Update app.py

app.py

@@ -34,8 +34,8 @@ MAX_FRAMES_MODEL = 81
 DEFAULT_NAG_NEGATIVE_PROMPT = "Static, motionless, still, ugly, bad quality, worst quality, poorly drawn, low resolution, blurry, lack of details"
 
 MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
-#SUB_MODEL_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
-#SUB_MODEL_FILENAME = "Wan14BT2VFusioniX_fp16_.safetensors"
+SUB_MODEL_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
+SUB_MODEL_FILENAME = "Wan14BT2VFusioniX_fp16_.safetensors"
 
 
 
@@ -45,8 +45,8 @@ MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
 
 # new experiment
 
-SUB_MODEL_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
-SUB_MODEL_FILENAME = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
+LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
+LORA_FILENAM = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
 
 vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 wan_path = hf_hub_download(repo_id=SUB_MODEL_ID, filename=SUB_MODEL_FILENAME)
@@ -58,6 +58,17 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow
 pipe.to("cuda")
 
 
+# Load FusionX enhancement LoRAs
+causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
+pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
+pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
+for name, param in pipe.transformer.named_parameters():
+   if "lora_B" in name:
+       if "blocks.0" in name:
+           param.data = param.data * 0.25
+pipe.fuse_lora()
+pipe.unload_lora_weights()
+
 
 pipe.transformer.__class__.attn_processors = NagWanTransformer3DModel.attn_processors
 pipe.transformer.__class__.set_attn_processor = NagWanTransformer3DModel.set_attn_processor