xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 00:16:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: update model ratio

Browse Source
This commit is contained in:
CaIon 2024-04-24 22:08:54 +08:00
parent 08b5336431
commit 86b17fcce8
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
common/model-ratio.go
View File

@ -105,6 +105,12 @@ var DefaultModelRatio = map[string]float64{
"yi-34b-chat-0205": 0.018, "yi-34b-chat-0205": 0.018,
"yi-34b-chat-200k": 0.0864, "yi-34b-chat-200k": 0.0864,
"yi-vl-plus": 0.0432, "yi-vl-plus": 0.0432,
"command": 0.5,
"command-nightly": 0.5,
"command-light": 0.5,
"command-light-nightly": 0.5,
"command-r": 0.25,
"command-r-plus ": 1.5,
} }
var DefaultModelPrice = map[string]float64{ var DefaultModelPrice = map[string]float64{
@ -225,7 +231,14 @@ func GetCompletionRatio(name string) float64 {
return 3 return 3
} }
if strings.HasPrefix(name, "command") { if strings.HasPrefix(name, "command") {
switch name {
case "command-r":
return 3
case "command-r-plus":
return 5 return 5
default:
return 2
}
} }
switch name { switch name {
case "llama2-70b-4096": case "llama2-70b-4096":