Add Nemotron to PP_SUPPORTED_MODELS (vllm-project#6863)

Signed-off-by: Alvant <[email protected]>
compressa-ai · Oct 26, 2024 · 3c35ae4 · 3c35ae4
1 parent 4543f58
commit 3c35ae4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vllm/config.py b/vllm/config.py
@@ -39,6 +39,7 @@
     "Phi3ForCausalLM",
     "GPT2LMHeadModel",
     "MixtralForCausalLM",
+    "NemotronForCausalLM",
 ]