Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-03-31 15:38:40 +02:00
committed by GitHub
parent 3b07ca78bb
commit a03cee7a1d
2 changed files with 68 additions and 0 deletions

View File

@@ -348,6 +348,42 @@ class Qwen3ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
self.model_tester = Qwen3ModelTester(self)
self.config_tester = ConfigTester(self, config_class=Qwen3Config, hidden_size=37)
@unittest.skip("TODO: ask the contributor to take a look")
def test_beam_search_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_matches_greedy_search_0_random(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_contrastive_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_matches_greedy_search_1_same(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_sample(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_dola_decoding_sample(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_greedy_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_prompt_lookup_decoding_matches_greedy_search(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_generate_compilation_all_outputs(self):
pass
def test_config(self):
self.config_tester.run_common_tests()

View File

@@ -367,6 +367,38 @@ class Qwen3MoeModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterM
self.model_tester = Qwen3MoeModelTester(self)
self.config_tester = ConfigTester(self, config_class=Qwen3MoeConfig, hidden_size=37)
@unittest.skip("TODO: ask the contributor to take a look")
def test_beam_search_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_matches_greedy_search_0_random(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_contrastive_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_matches_greedy_search_1_same(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_assisted_decoding_sample(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_dola_decoding_sample(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_greedy_generate_dict_outputs_use_cache(self):
pass
@unittest.skip("TODO: ask the contributor to take a look")
def test_prompt_lookup_decoding_matches_greedy_search(self):
pass
def test_config(self):
self.config_tester.run_common_tests()