From 742abebb39d01e699d0478d29eb46eb056f418bc Mon Sep 17 00:00:00 2001 From: teleprint-me <77757836+teleprint-me@users.noreply.github.com> Date: Fri, 17 May 2024 20:37:59 -0400 Subject: [PATCH] refactor: Add log for status and fix url path variable name --- gguf-py/gguf/huggingface_hub.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/gguf-py/gguf/huggingface_hub.py b/gguf-py/gguf/huggingface_hub.py index 4667ec829..a25a35041 100644 --- a/gguf-py/gguf/huggingface_hub.py +++ b/gguf-py/gguf/huggingface_hub.py @@ -4,6 +4,8 @@ import pathlib import requests +from .constants import TokenizerType + class HuggingFaceHub: def __init__(self, auth_token: None | str, logger: None | logging.Logger): @@ -51,7 +53,8 @@ class HuggingFaceHub: return f"{self._base_url}/{repo}/resolve/main/{file}" def download_file(self, repo: str, file: str): - endpoint = self.resolve_path(repo, file) - response = self._session.get(endpoint, headers=self.headers) + resolve_path = self.resolve_path(repo, file) + response = self._session.get(resolve_path, headers=self.headers) + self.logger.info(f"Response status was {response.status_code}") response.raise_for_status() return response