Add Nemotron to PP_SUPPORTED_MODELS (#6863)

This commit is contained in:
Michael Goin
2024-07-27 15:05:17 -07:00
committed by GitHub
parent 75acdaa4b6
commit b1366a9534
+1
View File
@@ -39,6 +39,7 @@ _PP_SUPPORTED_MODELS = [
"Phi3ForCausalLM",
"GPT2LMHeadModel",
"MixtralForCausalLM",
"NemotronForCausalLM",
]