feat: 增加重置模型倍率功能 (close #62)

This commit is contained in:
1808837298@qq.com committed 2024-05-24 15:28:16 +08:00
1 parent 222f5ec1e8
commit f723ff9ba0
9 files changed
+2811 -185

No files matched your search

+1 -1
View File
@@ -30,7 +30,7 @@ func InitTokenEncoders() {
if err != nil {
common.FatalLog(fmt.Sprintf("failed to get gpt-4 token encoder: %s", err.Error()))
}
for model, _ := range common.DefaultModelRatio {
for model, _ := range common.GetDefaultModelRatioMap() {
if strings.HasPrefix(model, "gpt-3.5") {
tokenEncoderMap[model] = gpt35TokenEncoder
} else if strings.HasPrefix(model, "gpt-4") {