Skip to content

GitLab

Unverified Commit 6c7335e0 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Jamba: fix left-padding test (#30389) 

fix test
parent f3b3533e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/jamba/test_modeling_jamba.py
View file @ 6c7335e0
...@@ -483,7 +483,7 @@ class JambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi ...@@ -483,7 +483,7 @@ class JambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
return model_kwargs return model_kwargs
for model_class in decoder_only_classes: 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() model = model_class(config).to(torch_device).eval()
signature = inspect.signature(model.forward).parameters.keys() signature = inspect.signature(model.forward).parameters.keys()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment