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
|