From 3c35ae449f2bf63129f1454118dafbd38d76a409 Mon Sep 17 00:00:00 2001
From: Michael Goin <michael@neuralmagic.com>
Date: Sat, 27 Jul 2024 18:05:17 -0400
Subject: [PATCH] Add Nemotron to PP_SUPPORTED_MODELS (#6863)

Signed-off-by: Alvant <alvasian@yandex.ru>
---
 vllm/config.py | 1 +
 1 file changed, 1 insertion(+)

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",
 ]