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
90e10dee
Commit
90e10dee
authored
Jun 09, 2025
by
zhuwenwen
Browse files
[tests] update models
parent
d7d4623e
Changes
2
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
188 additions
and
182 deletions
+188
-182
tests/models/registry.py
tests/models/registry.py
+184
-180
tests/models/test_gptq_bitblas.py
tests/models/test_gptq_bitblas.py
+4
-2
No files found.
tests/models/registry.py
View file @
90e10dee
This diff is collapsed.
Click to expand it.
tests/models/test_gptq_bitblas.py
View file @
90e10dee
...
@@ -13,9 +13,11 @@ Run `pytest tests/models/test_bitblas.py`.
...
@@ -13,9 +13,11 @@ Run `pytest tests/models/test_bitblas.py`.
"""
"""
from
dataclasses
import
dataclass
from
dataclasses
import
dataclass
import
os
import
pytest
import
pytest
from
.utils
import
check_logprobs_close
from
.utils
import
check_logprobs_close
from
..utils
import
models_path_prefix
@
dataclass
@
dataclass
...
@@ -24,7 +26,7 @@ class ModelPair:
...
@@ -24,7 +26,7 @@ class ModelPair:
model_pairs
=
[
model_pairs
=
[
ModelPair
(
model_gptq
=
"hxbgsyxh/opt-125m-4bit-128g"
),
ModelPair
(
model_gptq
=
os
.
path
.
join
(
models_path_prefix
,
"hxbgsyxh/opt-125m-4bit-128g"
)
)
,
]
]
...
@@ -58,4 +60,4 @@ def test_models(
...
@@ -58,4 +60,4 @@ def test_models(
outputs_1_lst
=
bitblas_outputs
,
outputs_1_lst
=
bitblas_outputs
,
name_0
=
"gptq"
,
name_0
=
"gptq"
,
name_1
=
"gptq_bitblas"
,
name_1
=
"gptq_bitblas"
,
)
)
\ No newline at end of file
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