ar08 commited on
Commit
afa5e5a
1 Parent(s): b11cfc4

Update shortGPT/gpt/gpt_utils.py

Browse files
Files changed (1) hide show
  1. shortGPT/gpt/gpt_utils.py +1 -1
shortGPT/gpt/gpt_utils.py CHANGED
@@ -10,7 +10,7 @@ import yaml
10
  from shortGPT.config.api_db import ApiKeyManager
11
 
12
 
13
- def num_tokens_from_messages(texts, model="gpt-3.5-turbo-0301"):
14
  """Returns the number of tokens used by a list of messages."""
15
  try:
16
  encoding = tiktoken.encoding_for_model(model)
 
10
  from shortGPT.config.api_db import ApiKeyManager
11
 
12
 
13
+ def num_tokens_from_messages(texts, model="cosmosrp"):
14
  """Returns the number of tokens used by a list of messages."""
15
  try:
16
  encoding = tiktoken.encoding_for_model(model)