Add Nemotron to PP_SUPPORTED_MODELS (#6863)

mgoin · web-flow · commit b1366a953498 · 2024-07-27T15:05:17.000-07:00
diff --git a/vllm/config.py b/vllm/config.py
@@ -39,6 +39,7 @@
     "Phi3ForCausalLM",
     "GPT2LMHeadModel",
     "MixtralForCausalLM",
+    "NemotronForCausalLM",
 ]
 
 
