dkolarova commited on
Commit
ebeff7a
·
verified ·
1 Parent(s): d4b3cd7

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -0
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