biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Model] Added GLM-4 series hf format model support vllm==0.6.4 (#10561)

Browse Source 
Signed-off-by: Isotr0py <2037008807@qq.com>
Co-authored-by: Isotr0py <2037008807@qq.com>
Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
This commit is contained in:
sixgod
2024-11-28 22:53:31 +08:00
committed by GitHub
parent 3ed5e73146
commit 5fc5ce0fe4
5 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/test_initialization.py
View File

@@ -11,7 +11,7 @@ from .registry import HF_EXAMPLE_MODELS
@pytest.mark.parametrize("model_arch", HF_EXAMPLE_MODELS.get_supported_archs())
def test_can_initialize(model_arch):
if (model_arch == "Idefics3ForConditionalGeneration"
if (model_arch in {"Idefics3ForConditionalGeneration", "GlmForCausalLM"}
and transformers.__version__ < "4.46.0"):
pytest.skip(reason="Model introduced in HF >= 4.46.0")