diff --git a/inference/cli_demo.py b/inference/cli_demo.py index 37dfcfc..752008e 100644 --- a/inference/cli_demo.py +++ b/inference/cli_demo.py @@ -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`.