Update shortGPT/gpt/gpt_utils.py
Browse files
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="
|
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)
|