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

[Model] Add FlexOlmo model implementation (#24923)

Browse Source 
Signed-off-by: Shane A <shanea@allenai.org>
This commit is contained in:
Shane A
2025-10-10 09:43:15 -07:00
committed by GitHub
parent b2155ed317
commit 8d2b8c0ff2
8 changed files with 286 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -90,6 +90,7 @@ _TEXT_GENERATION_MODELS = {
"Exaone4ForCausalLM": ("exaone4", "Exaone4ForCausalLM"),
"FalconForCausalLM": ("falcon", "FalconForCausalLM"),
"Fairseq2LlamaForCausalLM": ("fairseq2_llama", "Fairseq2LlamaForCausalLM"),
"FlexOlmoForCausalLM": ("flex_olmo", "FlexOlmoForCausalLM"),
"GemmaForCausalLM": ("gemma", "GemmaForCausalLM"),
"Gemma2ForCausalLM": ("gemma2", "Gemma2ForCausalLM"),
"Gemma3ForCausalLM": ("gemma3", "Gemma3ForCausalLM"),