From 04bffcea924700a236c261e280b3b6ea3dc410d4 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 15 Aug 2023 22:39:37 -0700 Subject: [PATCH] cleanup --- litellm/llms/huggingface_restapi.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 7b88d03e6..50bb9451a 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -67,10 +67,10 @@ class HuggingfaceRestAPILLM(): print_verbose(f"raw model_response: {response.text}") ## RESPONSE OBJECT completion_response = response.json() - print(f"response: {completion_response}") + print_verbose(f"response: {completion_response}") if isinstance(completion_response, dict) and "error" in completion_response: - print(f"completion error: {completion_response['error']}") - print(f"response.status_code: {response.status_code}") + print_verbose(f"completion error: {completion_response['error']}") + print_verbose(f"response.status_code: {response.status_code}") raise HuggingfaceError(message=completion_response["error"], status_code=response.status_code) else: model_response["choices"][0]["message"]["content"] = completion_response[0]["generated_text"]