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

patch: Apply minor fixes for handling headers and writing content

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

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

@ -4,14 +4,14 @@ import pathlib
import requests import requests
from .constants import TokenizerType from .constants import MODEL_REPOS, 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):
# Set headers if authentication is available # Set headers if authentication is available
if auth_token is None: if auth_token is None:
self._headers = {} self._headers = None
else: else:
self._headers = {"Authorization": f"Bearer {auth_token}"} self._headers = {"Authorization": f"Bearer {auth_token}"}
@ -43,11 +43,10 @@ class HuggingFaceHub:
def base_url(self) -> str: def base_url(self) -> str:
return self._base_url return self._base_url
@staticmethod def write_file(self, content: bytes, path: pathlib.Path) -> None:
def write_file(self, content: bytes, save_path: pathlib.Path) -> None: with open(path, 'wb') as f:
with open(save_path, 'wb') as f:
f.write(content) f.write(content)
self.logger.info(f"File {save_path} downloaded successfully") self.logger.info(f"Wrote {len(content)} bytes to {path} successfully")
def resolve_path(self, repo: str, file: str) -> str: def resolve_path(self, repo: str, file: str) -> str:
return f"{self._base_url}/{repo}/resolve/main/{file}" return f"{self._base_url}/{repo}/resolve/main/{file}"