rahul7star commited on
Commit
e289134
·
verified ·
1 Parent(s): 1ac1194

Update wan2_fast.py

Browse files
Files changed (1) hide show
  1. wan2_fast.py +14 -0
wan2_fast.py CHANGED
@@ -312,6 +312,20 @@ torch.cuda.empty_cache()
312
  # MIN_FRAMES_MODEL = 8
313
  # MAX_FRAMES_MODEL = 81
314
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
315
  # Constants
316
  MOD_VALUE = 32
317
  DEFAULT_H_SLIDER_VALUE = 896
 
312
  # MIN_FRAMES_MODEL = 8
313
  # MAX_FRAMES_MODEL = 81
314
 
315
+
316
+
317
+
318
+ LORA_REPO_ID = "rahul7star/wan2.2Lora""
319
+ LORA_FILENAME = "c0wg1rl.3_wan22-5b-ti2v - e380.safetensors"
320
+ causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
321
+ pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
322
+ pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
323
+ pipe.fuse_lora()
324
+
325
+
326
+
327
+
328
+
329
  # Constants
330
  MOD_VALUE = 32
331
  DEFAULT_H_SLIDER_VALUE = 896