testing new Lora

app.py

@@ -36,8 +36,17 @@ DEFAULT_NAG_NEGATIVE_PROMPT = "Static, motionless, still, ugly, bad quality, wor
 MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
 SUB_MODEL_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
 SUB_MODEL_FILENAME = "Wan14BT2VFusioniX_fp16_.safetensors"
-LORA_REPO_ID = "Kijai/WanVideo_comfy"
-LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
+
+
+
+#LORA_REPO_ID = "Kijai/WanVideo_comfy"
+#LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
+
+
+# new experiment
+
+LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
+LORA_FILENAME = "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)
@@ -48,15 +57,7 @@ pipe = NAGWanPipeline.from_pretrained(
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
 pipe.to("cuda")
 
-#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