Chranos/enginex-mlu370-vllm
1
0
Fork 0
forked from EngineX-Cambricon/enginex-mlu370-vllm
Code Pull Requests Activity

add gemma3

Browse Source
This commit is contained in:
Chranos
2026-02-10 13:00:25 +08:00
parent 40dee08f7b
commit 3ef89630ab
4 changed files with 465 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
vllm-v0.6.2/vllm/model_executor/models/registry.py
View File

@@ -52,6 +52,7 @@ _TEXT_GENERATION_MODELS = {
"FalconForCausalLM": ("falcon", "FalconForCausalLM"),
"GemmaForCausalLM": ("gemma", "GemmaForCausalLM"),
"Gemma2ForCausalLM": ("gemma2", "Gemma2ForCausalLM"),
"Gemma3ForCausalLM": ("gemma3", "Gemma3ForCausalLM"),
"GPT2LMHeadModel": ("gpt2", "GPT2LMHeadModel"),
"GPTBigCodeForCausalLM": ("gpt_bigcode", "GPTBigCodeForCausalLM"),
"GPTJForCausalLM": ("gpt_j", "GPTJForCausalLM"),