Add Nemotron to PP_SUPPORTED_MODELS (#6863)

This commit is contained in:
Michael Goin 2024-07-27 18:05:17 -04:00 committed by GitHub
parent 75acdaa4b6
commit b1366a9534
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -39,6 +39,7 @@ _PP_SUPPORTED_MODELS = [
"Phi3ForCausalLM",
"GPT2LMHeadModel",
"MixtralForCausalLM",
"NemotronForCausalLM",
]