rahul7star commited on
Commit
cb4e77b
·
verified ·
1 Parent(s): 6e5a754

Update app_two_lora.py

Browse files
Files changed (1) hide show
  1. app_two_lora.py +27 -6
app_two_lora.py CHANGED
@@ -14,7 +14,11 @@ import os
14
  # LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
15
 
16
 
17
- MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
 
 
 
 
18
 
19
 
20
 
@@ -26,12 +30,12 @@ MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
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"
34
- LORA_TWO_FILE="FusionX_FaceNaturalizer.safetensors"
35
 
36
 
37
  # LORA_TWO = "vrgamedevgirl84/Wan14BT2VFusioniX"
@@ -50,9 +54,13 @@ LORA_TWO_FILE="FusionX_FaceNaturalizer.safetensors"
50
 
51
 
52
 
 
53
 
54
 
 
 
55
 
 
56
 
57
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
58
  pipe = WanPipeline.from_pretrained(
@@ -61,7 +69,20 @@ pipe = WanPipeline.from_pretrained(
61
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)
62
  pipe.to("cuda")
63
 
64
- causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
 
 
 
 
 
 
 
 
 
 
 
 
 
65
 
66
 
67
 
 
14
  # LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
15
 
16
 
17
+ #MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
18
+
19
+
20
+
21
+
22
 
23
 
24
 
 
30
 
31
 
32
  #LORA_FILENAME = "Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors"
33
+ # LORA_REPO_ID = "moonshotmillion/Wan_FusionX_FaceNaturalizer"
34
+ # LORA_FILENAME = "FusionX_FaceNaturalizer.safetensors"
35
 
36
 
37
+ # LORA_TWO="moonshotmillion/Wan_FusionX_FaceNaturalizer"
38
+ # LORA_TWO_FILE="FusionX_FaceNaturalizer.safetensors"
39
 
40
 
41
  # LORA_TWO = "vrgamedevgirl84/Wan14BT2VFusioniX"
 
54
 
55
 
56
 
57
+ MODEL_ID = "Wan-AI/Wan2.1-VACE-14B-diffusers"
58
 
59
 
60
+ LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
61
+ LORA_FILENAME = "FusionX_LoRa/Phantom_Wan_14B_FusionX_LoRA.safetensors"
62
 
63
+ causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
64
 
65
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
66
  pipe = WanPipeline.from_pretrained(
 
69
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)
70
  pipe.to("cuda")
71
 
72
+
73
+
74
+
75
+
76
+ # pipe.load_lora_weights(
77
+ # "vrgamedevgirl84/Wan14BT2VFusioniX",
78
+ # weight_name="OtherLoRa's/DetailEnhancerV1.safetensors", adapter_name="detailer"
79
+ # )
80
+ # pipe.set_adapters(["phantom","detailer"], adapter_weights=[1, .9])
81
+ # pipe.fuse_lora()
82
+
83
+
84
+
85
+
86
 
87
 
88