mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-02 10:45:00 +00:00
catch litellm bug for image size
This commit is contained in:
parent
2bb75dc11f
commit
8d4d549a98
1 changed files with 5 additions and 1 deletions
|
@ -516,7 +516,11 @@ class Model:
|
|||
|
||||
def token_count(self, messages):
|
||||
if type(messages) is list:
|
||||
return litellm.token_counter(model=self.name, messages=messages)
|
||||
try:
|
||||
return litellm.token_counter(model=self.name, messages=messages)
|
||||
except Exception as err:
|
||||
print(f"Unable to count tokens: {err}")
|
||||
return 0
|
||||
|
||||
if not self.tokenizer:
|
||||
return
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue