diff --git a/tests/lora/test_worker.py b/tests/lora/test_worker.py index 30b74ce3ef70e..e5ae660af1400 100644 --- a/tests/lora/test_worker.py +++ b/tests/lora/test_worker.py @@ -58,13 +58,19 @@ def test_worker_apply_lora(sql_lora_files): download_dir=None, load_format="dummy", ), - parallel_config=ParallelConfig(1, 1, False), + parallel_config=ParallelConfig( + pipeline_parallel_size=1, + tensor_parallel_size=1, + data_parallel_size=1, + ), scheduler_config=SchedulerConfig("generate", 32, 32, 32), device_config=DeviceConfig("cuda"), - cache_config=CacheConfig(block_size=16, - gpu_memory_utilization=1., - swap_space=0, - cache_dtype="auto"), + cache_config=CacheConfig( + block_size=16, + gpu_memory_utilization=1.0, + swap_space=0, + cache_dtype="auto", + ), lora_config=LoRAConfig(max_lora_rank=8, max_cpu_loras=32, max_loras=32), )