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

Merge pull request #719 from holma91/fix-lora-scale

Browse Source 
fix scale bug
This commit is contained in:
OleehyO 2025-03-25 18:59:49 +08:00 committed by GitHub
commit 5ab1e2449f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -129,7 +129,7 @@ def generate_video(
pipe.load_lora_weights( pipe.load_lora_weights(
lora_path, weight_name="pytorch_lora_weights.safetensors", adapter_name="test_1" 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. # 2. Set Scheduler.
# Can be changed to `CogVideoXDPMScheduler` or `CogVideoXDDIMScheduler`. # Can be changed to `CogVideoXDPMScheduler` or `CogVideoXDDIMScheduler`.