Merge pull request #335 from google-research/rajat-dev

adapt to new huggingface_hub interface
This commit is contained in:
Rajat Sen
2025-10-31 10:45:25 -07:00
committed by GitHub
2 changed files with 32 additions and 23 deletions
+28 -19
View File
@@ -1,40 +1,49 @@
# This file was autogenerated by uv via the following command:
# uv pip compile pyproject.toml -o requirements.txt
anyio==4.11.0
# via httpx
certifi==2025.10.5
# via requests
charset-normalizer==3.4.3
# via requests
# via
# httpcore
# httpx
click==8.3.0
# via typer-slim
filelock==3.19.1
# via huggingface-hub
fsspec==2025.9.0
# via huggingface-hub
hf-xet==1.1.10
h11==0.16.0
# via httpcore
hf-xet==1.2.0
# via huggingface-hub
huggingface-hub==0.35.3
httpcore==1.0.9
# via httpx
httpx==0.28.1
# via huggingface-hub
huggingface-hub==1.0.1
# via timesfm (pyproject.toml)
idna==3.10
# via requests
inquirerpy==0.3.4
# via huggingface-hub
# via
# anyio
# httpx
numpy==2.2.6
# via timesfm (pyproject.toml)
packaging==25.0
# via huggingface-hub
pfzy==0.3.4
# via inquirerpy
prompt-toolkit==3.0.52
# via inquirerpy
pyyaml==6.0.3
# via huggingface-hub
requests==2.32.5
# via huggingface-hub
safetensors==0.6.2
# via timesfm (pyproject.toml)
shellingham==1.5.4
# via huggingface-hub
sniffio==1.3.1
# via anyio
tqdm==4.67.1
# via huggingface-hub
typing-extensions==4.15.0
typer-slim==0.20.0
# via huggingface-hub
urllib3==2.5.0
# via requests
wcwidth==0.2.14
# via prompt-toolkit
typing-extensions==4.15.0
# via
# anyio
# huggingface-hub
# typer-slim
+4 -4
View File
@@ -275,11 +275,11 @@ class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5, ModelHubMixin):
model_id: str = "google/timesfm-2.5-200m-pytorch",
revision: Optional[str],
cache_dir: Optional[Union[str, Path]],
force_download: bool,
proxies: Optional[Dict],
resume_download: Optional[bool],
force_download: bool = True,
proxies: Optional[Dict] = None,
resume_download: Optional[bool] = None,
local_files_only: bool,
token: Optional[str],
token: Optional[Union[str, bool]],
**model_kwargs,
):
"""