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

feat: update model ratio

Browse Source
This commit is contained in:
1808837298@qq.com 2024-05-08 16:57:11 +08:00
parent 2bf404507f
commit d7a343e2f6
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
common/model-ratio.go
View File

@ -111,6 +111,8 @@ var DefaultModelRatio = map[string]float64{
"command-light-nightly": 0.5, "command-light-nightly": 0.5,
"command-r": 0.25, "command-r": 0.25,
"command-r-plus ": 1.5, "command-r-plus ": 1.5,
"deepseek-chat": 0.07,
"deepseek-coder": 0.07,
} }
var DefaultModelPrice = map[string]float64{ var DefaultModelPrice = map[string]float64{
@ -240,6 +242,9 @@ func GetCompletionRatio(name string) float64 {
return 2 return 2
} }
} }
if strings.HasPrefix(name, "deepseek") {
return 2
}
switch name { switch name {
case "llama2-70b-4096": case "llama2-70b-4096":
return 0.8 / 0.7 return 0.8 / 0.7