Spaces:
Paused
Paused
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 |
|