Spaces:
Runtime error
Runtime error
Update app_two_lora.py
Browse files- 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.
|
|
|
|
|
|
|
|
|
|
|
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 |
|