diff --git a/tests/test_hub.py b/tests/test_hub.py index 0c31bc7..56ae85b 100644 --- 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]):