101 references to ModelEncoding
Microsoft.ML.Tokenizers (101)
Model\TiktokenTokenizer.cs (101)
1040private static readonly (string Prefix, ModelEncoding Encoding)[] _modelPrefixToEncoding = 1042( "o1-", ModelEncoding.O200kBase ), // e.g. o1-mini 1043( "o3-", ModelEncoding.O200kBase ), // e.g. o3-mini 1044( "o4-mini-", ModelEncoding.O200kBase ), // e.g. o4-mini 1047( "gpt-5.6-", ModelEncoding.O200kBase ), 1048( "gpt-5.5-", ModelEncoding.O200kBase ), 1049( "gpt-5.4-", ModelEncoding.O200kBase ), 1050( "gpt-5.3-", ModelEncoding.O200kBase ), 1051( "gpt-5.2-", ModelEncoding.O200kBase ), 1052( "gpt-5.1-", ModelEncoding.O200kBase ), 1053( "gpt-5-", ModelEncoding.O200kBase ), 1054( "gpt-4.1-", ModelEncoding.O200kBase ), // e.g., gpt-4.1-mini 1055( "gpt-4.5-", ModelEncoding.O200kBase ), // e.g., gpt-4.5 1056( "gpt-4o-", ModelEncoding.O200kBase ), // e.g., gpt-4o-2024-05-13 1057( "chatgpt-4o-", ModelEncoding.O200kBase ), 1058( "gpt-4-", ModelEncoding.Cl100kBase ), // e.g., gpt-4-0314, etc., plus gpt-4-32k 1059( "gpt-3.5-", ModelEncoding.Cl100kBase ), // e.g, gpt-3.5-turbo-0301, -0401, etc. 1060( "gpt-35-", ModelEncoding.Cl100kBase ), // Azure deployment name 1061( "gpt-oss-", ModelEncoding.O200kHarmony ), 1064( "ft:gpt-4o", ModelEncoding.O200kBase ), 1065( "ft:gpt-4", ModelEncoding.Cl100kBase ), 1066( "ft:gpt-3.5-turbo", ModelEncoding.Cl100kBase ), 1067( "ft:davinci-002", ModelEncoding.Cl100kBase ), 1068( "ft:babbage-002", ModelEncoding.Cl100kBase ), 1071private static readonly Dictionary<string, ModelEncoding> _modelToEncoding = 1072new Dictionary<string, ModelEncoding>(StringComparer.OrdinalIgnoreCase) 1075{ "o1", ModelEncoding.O200kBase }, 1076{ "o3", ModelEncoding.O200kBase }, 1077{ "o4-mini", ModelEncoding.O200kBase }, 1080{ "gpt-5.6", ModelEncoding.O200kBase }, 1081{ "gpt-5.5", ModelEncoding.O200kBase }, 1082{ "gpt-5.4", ModelEncoding.O200kBase }, 1083{ "gpt-5.3", ModelEncoding.O200kBase }, 1084{ "gpt-5.2", ModelEncoding.O200kBase }, 1085{ "gpt-5.1", ModelEncoding.O200kBase }, 1086{ "gpt-5", ModelEncoding.O200kBase }, 1087{ "gpt-4.1", ModelEncoding.O200kBase }, 1088{ "gpt-4o", ModelEncoding.O200kBase }, 1089{ "gpt-4", ModelEncoding.Cl100kBase }, 1090{ "gpt-3.5-turbo", ModelEncoding.Cl100kBase }, 1091{ "gpt-3.5", ModelEncoding.Cl100kBase }, 1092{ "gpt-3.5-turbo-16k", ModelEncoding.Cl100kBase }, 1093{ "gpt-35", ModelEncoding.Cl100kBase }, // Azure deployment name 1094{ "gpt-35-turbo", ModelEncoding.Cl100kBase }, // Azure deployment name 1095{ "gpt-35-turbo-16k", ModelEncoding.Cl100kBase }, // Azure deployment name 1098{ "davinci-002", ModelEncoding.Cl100kBase }, 1099{ "babbage-002", ModelEncoding.Cl100kBase }, 1103{ "text-embedding-ada-002", ModelEncoding.Cl100kBase }, 1104{ "text-embedding-3-small", ModelEncoding.Cl100kBase }, 1105{ "text-embedding-3-large", ModelEncoding.Cl100kBase }, 1109{ "text-davinci-003", ModelEncoding.P50kBase }, 1110{ "text-davinci-002", ModelEncoding.P50kBase }, 1111{ "text-davinci-001", ModelEncoding.R50kBase }, 1112{ "text-curie-001", ModelEncoding.R50kBase }, 1113{ "text-babbage-001", ModelEncoding.R50kBase }, 1114{ "text-ada-001", ModelEncoding.R50kBase }, 1115{ "davinci", ModelEncoding.R50kBase }, 1116{ "curie", ModelEncoding.R50kBase }, 1117{ "babbage", ModelEncoding.R50kBase }, 1118{ "ada", ModelEncoding.R50kBase }, 1121{ "code-davinci-002", ModelEncoding.P50kBase }, 1122{ "code-davinci-001", ModelEncoding.P50kBase }, 1123{ "code-cushman-002", ModelEncoding.P50kBase }, 1124{ "code-cushman-001", ModelEncoding.P50kBase }, 1125{ "davinci-codex", ModelEncoding.P50kBase }, 1126{ "cushman-codex", ModelEncoding.P50kBase }, 1129{ "text-davinci-edit-001", ModelEncoding.P50kEdit }, 1130{ "code-davinci-edit-001", ModelEncoding.P50kEdit }, 1134{ "text-similarity-davinci-001", ModelEncoding.R50kBase }, 1135{ "text-similarity-curie-001", ModelEncoding.R50kBase }, 1136{ "text-similarity-babbage-001", ModelEncoding.R50kBase }, 1137{ "text-similarity-ada-001", ModelEncoding.R50kBase }, 1138{ "text-search-davinci-doc-001", ModelEncoding.R50kBase }, 1139{ "text-search-curie-doc-001", ModelEncoding.R50kBase }, 1140{ "text-search-babbage-doc-001", ModelEncoding.R50kBase }, 1141{ "text-search-ada-doc-001", ModelEncoding.R50kBase }, 1142{ "code-search-babbage-code-001", ModelEncoding.R50kBase }, 1143{ "code-search-ada-code-001", ModelEncoding.R50kBase }, 1146{ "gpt2", ModelEncoding.GPT2 }, 1147{ "gpt-2", ModelEncoding.GPT2 }, 1150{ Phi4ModelName, ModelEncoding.Cl100kBase }, 1153private static ModelEncoding GetModelEncoding(string modelName) 1155if (!_modelToEncoding.TryGetValue(modelName, out ModelEncoding encoder)) 1157foreach ((string Prefix, ModelEncoding Encoding) in _modelPrefixToEncoding) 1167if (encoder == ModelEncoding.None) 1203private static (Dictionary<string, int> SpecialTokens, Regex Regex, string VocabFile, Type? DataType, string PackageName) GetTiktokenConfigurations(ModelEncoding modelEncoding, string? modelName = null) 1207case ModelEncoding.Cl100kBase: 1216case ModelEncoding.GPT2: 1219case ModelEncoding.O200kBase: 1222case ModelEncoding.P50kBase: 1225case ModelEncoding.P50kEdit: 1229case ModelEncoding.R50kBase: 1232case ModelEncoding.O200kHarmony: 1298ModelEncoding modelEncoding, 1544ModelEncoding modelEncoding; 1547modelEncoding = ModelEncoding.Cl100kBase; 1551modelEncoding = ModelEncoding.O200kBase; 1555modelEncoding = ModelEncoding.O200kHarmony; 1559modelEncoding = ModelEncoding.P50kBase; 1563modelEncoding = ModelEncoding.P50kEdit; 1567modelEncoding = ModelEncoding.R50kBase;