Add Nemotron to PP_SUPPORTED_MODELS

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