|
@@ -22,12 +22,16 @@ from core.model_runtime.utils import helper
|
|
|
class NVIDIALargeLanguageModel(OAIAPICompatLargeLanguageModel):
|
|
|
MODEL_SUFFIX_MAP = {
|
|
|
'fuyu-8b': 'vlm/adept/fuyu-8b',
|
|
|
+ 'mistralai/mistral-large': '',
|
|
|
'mistralai/mixtral-8x7b-instruct-v0.1': '',
|
|
|
+ 'mistralai/mixtral-8x22b-instruct-v0.1': '',
|
|
|
'google/gemma-7b': '',
|
|
|
'google/codegemma-7b': '',
|
|
|
+ 'snowflake/arctic':'',
|
|
|
'meta/llama2-70b': '',
|
|
|
'meta/llama3-8b-instruct': '',
|
|
|
- 'meta/llama3-70b-instruct': ''
|
|
|
+ 'meta/llama3-70b-instruct': '',
|
|
|
+ 'google/recurrentgemma-2b': ''
|
|
|
|
|
|
}
|
|
|
|