rahul7star commited on
Commit
48cc072
·
verified ·
1 Parent(s): dd38515

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -11
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
- # Load FusionX enhancement LoRAs
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