Add mapping for ewiz specific model

NREL · Jul 19, 2024 · 09a2bee · 09a2bee
1 parent 1c9cf0d
commit 09a2bee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elm/base.py b/elm/base.py
@@ -357,8 +357,8 @@ def count_tokens(text, model):
 
         # Optional mappings for weird azure names to tiktoken/openai names
         tokenizer_aliases = {'gpt-35-turbo': 'gpt-3.5-turbo',
-                             'gpt-4-32k': 'gpt-4-32k-0314'
-                             }
+                             'gpt-4-32k': 'gpt-4-32k-0314',
+                             'ewiz-gpt-4': 'gpt-4'}
 
         token_model = tokenizer_aliases.get(model, model)
         encoding = tiktoken.encoding_for_model(token_model)