jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix-nvidia-llama3 (#3973)

Browse source
This commit is contained in:
Joshua 2024-04-29 13:41:15 +08:00 • committed by GitHub
commit 2f84d00300
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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': ''
} }