Spaces:
Sleeping
Sleeping
Update models.py
Browse files
models.py
CHANGED
@@ -406,6 +406,7 @@ class HfApiModel(Model):
|
|
406 |
**kwargs,
|
407 |
)
|
408 |
response = self.client.chat_completion(**completion_kwargs)
|
|
|
409 |
|
410 |
self.last_input_token_count = response.usage.prompt_tokens
|
411 |
self.last_output_token_count = response.usage.completion_tokens
|
|
|
406 |
**kwargs,
|
407 |
)
|
408 |
response = self.client.chat_completion(**completion_kwargs)
|
409 |
+
logger.debug(response)
|
410 |
|
411 |
self.last_input_token_count = response.usage.prompt_tokens
|
412 |
self.last_output_token_count = response.usage.completion_tokens
|