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
e1109306
Unverified
Commit
e1109306
authored
Jun 27, 2025
by
Thomas Parnell
Committed by
GitHub
Jun 26, 2025
Browse files
[Fix] Fix gemma CI test failing on main (#20124)
Signed-off-by:
Thomas Parnell
<
tpa@zurich.ibm.com
>
parent
8b64c895
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
18 additions
and
11 deletions
+18
-11
tests/models/language/generation/test_gemma.py
tests/models/language/generation/test_gemma.py
+18
-11
No files found.
tests/models/language/generation/test_gemma.py
View file @
e1109306
...
...
@@ -7,14 +7,21 @@ MODELS = ["google/gemma-2b", "google/gemma-2-2b", "google/gemma-3-4b-it"]
@
pytest
.
mark
.
parametrize
(
"model"
,
MODELS
)
def
test_dummy_loader
(
vllm_runner
,
model
:
str
)
->
None
:
def
test_dummy_loader
(
vllm_runner
,
monkeypatch
,
model
:
str
)
->
None
:
with
monkeypatch
.
context
()
as
m
:
m
.
setenv
(
"VLLM_ALLOW_INSECURE_SERIALIZATION"
,
"1"
)
with
vllm_runner
(
model
,
load_format
=
"dummy"
,
)
as
llm
:
normalizers
=
llm
.
collective_rpc
(
lambda
self
:
self
.
worker
.
model_runner
.
model
.
model
.
normalizer
.
cpu
().
item
())
assert
np
.
allclose
(
normalizers
,
llm
.
llm_engine
.
model_config
.
hf_config
.
hidden_size
**
0.5
,
rtol
=
1e-3
)
if
model
==
"google/gemma-3-4b-it"
:
normalizers
=
llm
.
model
.
collective_rpc
(
lambda
self
:
self
.
model_runner
.
model
.
language_model
.
model
.
normalizer
.
cpu
().
item
())
config
=
llm
.
model
.
llm_engine
.
model_config
.
hf_config
.
text_config
else
:
normalizers
=
llm
.
model
.
collective_rpc
(
lambda
self
:
self
.
model_runner
.
model
.
model
.
normalizer
.
cpu
(
).
item
())
config
=
llm
.
model
.
llm_engine
.
model_config
.
hf_config
assert
np
.
allclose
(
normalizers
,
config
.
hidden_size
**
0.5
,
rtol
=
2e-3
)
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