File tree Expand file tree Collapse file tree 1 file changed +9
-6
lines changed
Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Original file line number Diff line number Diff line change @@ -74,8 +74,8 @@ package tokenizer
7474//}
7575
7676import (
77- "strings"
7877 "errors"
78+ "strings"
7979
8080 "github.com/tiktoken-go/tokenizer/codec"
8181)
@@ -144,11 +144,14 @@ const (
144144)
145145
146146var modelPrefixToEncoding map [Model ]Encoding = map [Model ]Encoding {
147- "o1-" : O200kBase ,
148- "gpt-4o-" : O200kBase ,
149- "gpt-4-" : Cl100kBase ,
150- "gpt-3.5-turbo-" : Cl100kBase ,
151- "gpt-35-turbo-" : Cl100kBase ,
147+ "o1-" : O200kBase ,
148+ // chat
149+ "chatgpt-4o-" : O200kBase ,
150+ "gpt-4o-" : O200kBase ,
151+ "gpt-4-" : Cl100kBase ,
152+ "gpt-3.5-turbo-" : Cl100kBase ,
153+ "gpt-35-turbo-" : Cl100kBase ,
154+ // fine-tuned
152155 "ft:gpt-4" : Cl100kBase ,
153156 "ft:gpt-3.5-turbo" : Cl100kBase ,
154157 "ft:davinci-002" : Cl100kBase ,
You can’t perform that action at this time.
0 commit comments