SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[qwen3] fix generation tests (#37142)

Browse Source 
* do not skip tests

* fix qwen3-moe as well

* fixup

* fixup
This commit is contained in:
Raushan Turganbay
2025-03-31 16:33:41 +02:00
committed by GitHub
parent e686fed635
commit 8805600406
6 changed files with 6 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/mixtral/test_modeling_mixtral.py
View File

@@ -351,7 +351,6 @@ class MixtralModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
def test_Mixtral_sequence_classification_model(self):
config, input_dict = self.model_tester.prepare_config_and_inputs_for_common()
print(config)
config.num_labels = 3
input_ids = input_dict["input_ids"]
attention_mask = input_ids.ne(1).to(torch_device)