Add Nemotron to PP_SUPPORTED_MODELS (#6863)

vllm-project · Jul 27, 2024 · b1366a9 · b1366a9
1 parent 75acdaa
commit b1366a9
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",
 ]