xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-11-15 02:13:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add gpt-4o tokenizer

Browse Source 
Signed-off-by: wozulong <>
This commit is contained in:
wozulong
2024-05-14 14:47:57 +08:00
parent 4f6c171a08
commit 7a7a923504
9 changed files with 32 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
common/model-ratio.go
View File

@@ -16,6 +16,7 @@ import (
var DefaultModelRatio = map[string]float64{
//"midjourney": 50,
"gpt-4-gizmo-*": 15,
"g-*": 15,
"gpt-4": 15,
"gpt-4-0314": 15,
"gpt-4-0613": 15,
@@ -120,6 +121,7 @@ var DefaultModelRatio = map[string]float64{
var DefaultModelPrice = map[string]float64{
"gpt-4-gizmo-*": 0.1,
"g-*": 0.1,
"mj_imagine": 0.1,
"mj_variation": 0.1,
"mj_reroll": 0.1,
@@ -162,6 +164,8 @@ func GetModelPrice(name string, printErr bool) float64 {
}
if strings.HasPrefix(name, "gpt-4-gizmo") {
name = "gpt-4-gizmo-*"
} else if strings.HasPrefix(name, "g-") {
name = "g-*"
}
price, ok := modelPrice[name]
if !ok {
@@ -195,6 +199,8 @@ func GetModelRatio(name string) float64 {
}
if strings.HasPrefix(name, "gpt-4-gizmo") {
name = "gpt-4-gizmo-*"
} else if strings.HasPrefix(name, "g-") {
name = "g-*"
}
ratio, ok := modelRatio[name]
if !ok {