Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
6c7335e0
Unverified
Commit
6c7335e0
authored
Apr 22, 2024
by
Joao Gante
Committed by
GitHub
Apr 22, 2024
Browse files
Jamba: fix left-padding test (#30389)
fix test
parent
f3b3533e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/models/jamba/test_modeling_jamba.py
tests/models/jamba/test_modeling_jamba.py
+1
-1
No files found.
tests/models/jamba/test_modeling_jamba.py
View file @
6c7335e0
...
...
@@ -483,7 +483,7 @@ class JambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
return
model_kwargs
for
model_class
in
decoder_only_classes
:
config
,
input_ids
,
attention_mask
,
_
=
self
.
_get_input_ids_and_config
()
config
,
input_ids
,
attention_mask
=
self
.
_get_input_ids_and_config
()
model
=
model_class
(
config
).
to
(
torch_device
).
eval
()
signature
=
inspect
.
signature
(
model
.
forward
).
parameters
.
keys
()
...
...
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