Add nvidia codegemma 7b (#3437)

This commit is contained in:
Josh Feng 2024-04-13 13:10:32 +08:00 committed by GitHub
parent a355225a83
commit 3e6631312d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 32 additions and 0 deletions

View File

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

View File

@ -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

View File

@ -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': ''
}