diff --git a/vllm/config.py b/vllm/config.py index 92fde449b43fd..e7b54e04b00d5 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -39,6 +39,7 @@ "Phi3ForCausalLM", "GPT2LMHeadModel", "MixtralForCausalLM", + "NemotronForCausalLM", ]