feat(qwen): add some more tests

huggingface · Jan 27, 2025 · 5dfd5a4 · 5dfd5a4
1 parent 2613bfa
commit 5dfd5a4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_hub.py b/tests/test_hub.py
@@ -132,6 +132,8 @@ def _reload():
         ("gemma", "GemmaForCausalLM"),
         ("mistral", "MistralForCausalLM"),
         ("mixtral", "MixtralForCausalLM"),
+        ("qwen", "QwenForCausalLM"),
+        ("qwen2", "QwenForCausalLM"),
     ),
 )
 def test_model_type_from_known_config(type: Tuple[str, str]):