From daf32f3bd4a364d171a52c633a53f9092ea29387 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Dec 2023 09:14:07 +0530 Subject: [PATCH] (fix) tg AI cost tracking - zero-one-ai/Yi-34B-Chat --- litellm/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/utils.py b/litellm/utils.py index 39723144889..4a983f9be66 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2350,6 +2350,7 @@ def client(original_function): def get_model_params_and_category(model_name): import re + model_name = model_name.lower() params_match = re.search( r"(\d+b)", model_name ) # catch all decimals like 3b, 70b, etc