phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) tg AI cost tracking - zero-one-ai/Yi-34B-Chat

Browse source
This commit is contained in:
ishaan-jaff 2023-12-29 09:14:07 +05:30
parent d79df3a1e9
commit daf32f3bd4
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/utils.py
View file

@ -2350,6 +2350,7 @@ def client(original_function):
def get_model_params_and_category(model_name): def get_model_params_and_category(model_name):
import re import re
model_name = model_name.lower()
params_match = re.search( params_match = re.search(
r"(\d+b)", model_name r"(\d+b)", model_name
) # catch all decimals like 3b, 70b, etc ) # catch all decimals like 3b, 70b, etc