Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ SUB_MODEL_FILENAME = "Wan14BT2VFusioniX_fp16_.safetensors"
|
|
43 |
#LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
|
44 |
|
45 |
|
46 |
-
# new experiment
|
47 |
|
48 |
LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
|
49 |
LORA_FILENAME = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
|
@@ -58,16 +58,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow
|
|
58 |
pipe.to("cuda")
|
59 |
|
60 |
|
61 |
-
|
62 |
-
causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
|
63 |
-
pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
|
64 |
-
pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
|
65 |
-
for name, param in pipe.transformer.named_parameters():
|
66 |
-
if "lora_B" in name:
|
67 |
-
if "blocks.0" in name:
|
68 |
-
param.data = param.data * 0.25
|
69 |
-
pipe.fuse_lora()
|
70 |
-
pipe.unload_lora_weights()
|
71 |
|
72 |
|
73 |
pipe.transformer.__class__.attn_processors = NagWanTransformer3DModel.attn_processors
|
|
|
43 |
#LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
|
44 |
|
45 |
|
46 |
+
# new experiment for future work
|
47 |
|
48 |
LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
|
49 |
LORA_FILENAME = "FusionX_LoRa/Wan2.1_T2V_14B_FusionX_LoRA.safetensors"
|
|
|
58 |
pipe.to("cuda")
|
59 |
|
60 |
|
61 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
|
63 |
|
64 |
pipe.transformer.__class__.attn_processors = NagWanTransformer3DModel.attn_processors
|