Spaces:
Paused
Paused
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
|