rahul7star commited on
Commit
2a59645
·
verified ·
1 Parent(s): 7a82290

Update app_two_lora.py

Browse files
Files changed (1) hide show
  1. app_two_lora.py +14 -3
app_two_lora.py CHANGED
@@ -19,9 +19,15 @@ MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
19
 
20
 
21
 
22
- LORA_REPO_ID = "Kijai/WanVideo_comfy"
23
- LORA_FILENAME = "Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank8_bf16.safetensors"
 
 
 
 
24
  #LORA_FILENAME = "Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors"
 
 
25
 
26
 
27
  LORA_TWO="moonshotmillion/Wan_FusionX_FaceNaturalizer"
@@ -66,7 +72,12 @@ causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
66
  lora2 = hf_hub_download(repo_id=LORA_TWO, filename=LORA_TWO_FILE)
67
 
68
  pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
69
- pipe.load_lora_weights(lora2 , adapter_name="nsfw_lora")
 
 
 
 
 
70
  pipe.set_adapters(["causvid_lora", "nsfw_lora"], adapter_weights=[0.30, 0.85])
71
  pipe.fuse_lora()
72
 
 
19
 
20
 
21
 
22
+ # LORA_REPO_ID = "Kijai/WanVideo_comfy"
23
+ # LORA_FILENAME = "Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank8_bf16.safetensors"
24
+
25
+
26
+
27
+
28
  #LORA_FILENAME = "Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors"
29
+ LORA_REPO_ID = "moonshotmillion/Wan_FusionX_FaceNaturalizer"
30
+ LORA_FILENAME = "FusionX_FaceNaturalizer.safetensors"
31
 
32
 
33
  LORA_TWO="moonshotmillion/Wan_FusionX_FaceNaturalizer"
 
72
  lora2 = hf_hub_download(repo_id=LORA_TWO, filename=LORA_TWO_FILE)
73
 
74
  pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
75
+ pipe.set_adapters(["causvid_lora"], adapter_weights=[0.30])
76
+
77
+
78
+
79
+ #pipe.load_lora_weights(lora2 , adapter_name="nsfw_lora")
80
+ #pipe.set_adapters(["causvid_lora", "nsfw_lora"], adapter_weights=[0.30, 0.85])
81
  pipe.set_adapters(["causvid_lora", "nsfw_lora"], adapter_weights=[0.30, 0.85])
82
  pipe.fuse_lora()
83