JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-05 11:28:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix scale bug

Browse Source
This commit is contained in:
holma91 2025-02-24 20:08:27 +01:00
parent e44c9f2c83
commit 84766d02e8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference/cli_demo.py
View File

@ -125,7 +125,7 @@ def generate_video(
# If you're using with lora, add this code
if lora_path:
pipe.load_lora_weights(lora_path, weight_name="pytorch_lora_weights.safetensors", adapter_name="test_1")
pipe.fuse_lora(components=["transformer"], lora_scale=1 / lora_rank)
pipe.fuse_lora(components=["transformer"], lora_scale=1.0)
# 2. Set Scheduler.
# Can be changed to `CogVideoXDPMScheduler` or `CogVideoXDDIMScheduler`.