vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Add log for status and fix url path variable name

Browse source
This commit is contained in:
teleprint-me 2024-05-17 20:37:59 -04:00
parent ba13d64bb3
commit 742abebb39
No known key found for this signature in database
GPG key ID: B0D11345E65C4D48
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
gguf-py/gguf/huggingface_hub.py
View file

@ -4,6 +4,8 @@ import pathlib
import requests import requests
from .constants import TokenizerType
class HuggingFaceHub: class HuggingFaceHub:
def __init__(self, auth_token: None | str, logger: None | logging.Logger): 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}" return f"{self._base_url}/{repo}/resolve/main/{file}"
def download_file(self, repo: str, file: str): def download_file(self, repo: str, file: str):
endpoint = self.resolve_path(repo, file) resolve_path = self.resolve_path(repo, file)
response = self._session.get(endpoint, headers=self.headers) response = self._session.get(resolve_path, headers=self.headers)
self.logger.info(f"Response status was {response.status_code}")
response.raise_for_status() response.raise_for_status()
return response return response