Spaces:
Running
on
Zero
Running
on
Zero
Update wan2_fast.py
Browse files- 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
|