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

[Model] Add Grok-2 (#31847)

Browse Source 
Signed-off-by: dangoldbj <dangoldbj23@gmail.com>
This commit is contained in:
Bijaya Dangol
2026-01-08 13:59:48 +01:00
committed by GitHub
parent 18d4e481d0
commit 59d260f5e4
8 changed files with 777 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/models/registry.py
View File

@@ -119,7 +119,8 @@ _TEXT_GENERATION_MODELS = {
"GraniteMoeHybridForCausalLM": ("granitemoehybrid", "GraniteMoeHybridForCausalLM"), # noqa: E501
"GraniteMoeSharedForCausalLM": ("granitemoeshared", "GraniteMoeSharedForCausalLM"), # noqa: E501
"GritLM": ("gritlm", "GritLM"),
"Grok1ModelForCausalLM": ("grok1", "Grok1ForCausalLM"),
"Grok1ModelForCausalLM": ("grok1", "GrokForCausalLM"),
"Grok1ForCausalLM": ("grok1", "GrokForCausalLM"),
"HunYuanMoEV1ForCausalLM": ("hunyuan_v1", "HunYuanMoEV1ForCausalLM"),
"HunYuanDenseV1ForCausalLM": ("hunyuan_v1", "HunYuanDenseV1ForCausalLM"),
"HCXVisionForCausalLM": ("hyperclovax_vision", "HCXVisionForCausalLM"),