estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix-nvidia-llama3 (#3973)

Browse Source
This commit is contained in:
Joshua
2024-04-29 13:41:15 +08:00
committed by GitHub
parent b82a2d97ef
commit 2f84d00300
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/model_runtime/model_providers/nvidia/llm/_position.yaml
View File

@@ -1,7 +1,7 @@
- google/gemma-7b - google/gemma-7b
- google/codegemma-7b - google/codegemma-7b
- meta/llama2-70b - meta/llama2-70b
- meta/llama3-8b - meta/llama3-8b-instruct
- meta/llama3-70b - meta/llama3-70b-instruct
- mistralai/mixtral-8x7b-instruct-v0.1 - mistralai/mixtral-8x7b-instruct-v0.1
- fuyu-8b - fuyu-8b

6
api/core/model_runtime/model_providers/nvidia/llm/llama3-70b.yaml
View File

@@ -1,7 +1,7 @@
model: meta/llama3-70b model: meta/llama3-70b-instruct
label: label:
zh_Hans: meta/llama3-70b zh_Hans: meta/llama3-70b-instruct
en_US: meta/llama3-70b en_US: meta/llama3-70b-instruct
model_type: llm model_type: llm
features: features:
- agent-thought - agent-thought

6
api/core/model_runtime/model_providers/nvidia/llm/llama3-8b.yaml
View File

@@ -1,7 +1,7 @@
model: meta/llama3-8b model: meta/llama3-8b-instruct
label: label:
zh_Hans: meta/llama3-8b zh_Hans: meta/llama3-8b-instruct
en_US: meta/llama3-8b en_US: meta/llama3-8b-instruct
model_type: llm model_type: llm
features: features:
- agent-thought - agent-thought

4
api/core/model_runtime/model_providers/nvidia/llm/llm.py
View File

@@ -26,8 +26,8 @@ class NVIDIALargeLanguageModel(OAIAPICompatLargeLanguageModel):
'google/gemma-7b': '', 'google/gemma-7b': '',
'google/codegemma-7b': '', 'google/codegemma-7b': '',
'meta/llama2-70b': '', 'meta/llama2-70b': '',
'meta/llama3-8b': '', 'meta/llama3-8b-instruct': '',
'meta/llama3-70b': '' 'meta/llama3-70b-instruct': ''
} }