diff --git a/vllm/config.py b/vllm/config.py index 92fde449..e7b54e04 100644 --- a/vllm/config.py +++ b/vllm/config.py @@ -39,6 +39,7 @@ _PP_SUPPORTED_MODELS = [ "Phi3ForCausalLM", "GPT2LMHeadModel", "MixtralForCausalLM", + "NemotronForCausalLM", ]