Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
259127f8
Unverified
Commit
259127f8
authored
May 14, 2025
by
Jee Jee Li
Committed by
GitHub
May 14, 2025
Browse files
[Bugfix] Fix LoRA test (#18123)
Signed-off-by:
Jee Jee Li
<
pandaleefree@gmail.com
>
parent
612c2edb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
5 deletions
+11
-5
tests/lora/test_worker.py
tests/lora/test_worker.py
+11
-5
No files found.
tests/lora/test_worker.py
View file @
259127f8
...
...
@@ -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
),
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment