Skip to content

Commit

Permalink
Add nvidia codegemma 7b (langgenius#3437)
Browse files Browse the repository at this point in the history
  • Loading branch information
joshfeng authored Apr 13, 2024
1 parent a355225 commit 3e66313
Show file tree
Hide file tree
Showing 3 changed files with 32 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
- google/gemma-7b
- google/codegemma-7b
- meta/llama2-70b
- mistralai/mixtral-8x7b-instruct-v0.1
- fuyu-8b
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
model: google/codegemma-7b
label:
zh_Hans: google/codegemma-7b
en_US: google/codegemma-7b
model_type: llm
features:
- agent-thought
model_properties:
mode: chat
context_size: 8192
parameter_rules:
- name: temperature
use_template: temperature
- name: top_p
use_template: top_p
- name: max_tokens
use_template: max_tokens
default: 1024
min: 1
max: 1024
- name: frequency_penalty
use_template: frequency_penalty
min: -2
max: 2
default: 0
- name: presence_penalty
use_template: presence_penalty
min: -2
max: 2
default: 0
1 change: 1 addition & 0 deletions api/core/model_runtime/model_providers/nvidia/llm/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ class NVIDIALargeLanguageModel(OAIAPICompatLargeLanguageModel):
'fuyu-8b': 'vlm/adept/fuyu-8b',
'mistralai/mixtral-8x7b-instruct-v0.1': '',
'google/gemma-7b': '',
'google/codegemma-7b': '',
'meta/llama2-70b': ''
}

Expand Down

0 comments on commit 3e66313

Please sign in to comment.