freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update timesfm_2p5_torch.py

Browse Source 
Update Hugging Face repo id.
This commit is contained in:
Yichen Zhou
2025-09-11 18:17:14 -07:00
committed by GitHub
parent 24dba82664
commit 59d2985681
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/timesfm/timesfm_2p5/timesfm_2p5_torch.py
+2 -2
View File
@@ -264,13 +264,13 @@ class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5):
self, self,
*, *,
path: str | None = None, path: str | None = None,
hf_repo_id: str | None = "gg-hf/timesfm-2.5-200m-pytorch", hf_repo_id: str | None = "google/timesfm-2.5-200m-pytorch",
): ):
"""Loads a PyTorch safetensors TimesFM model.""" """Loads a PyTorch safetensors TimesFM model."""
if path: if path:
pass pass
elif hf_repo_id: elif hf_repo_id:
logging.info("Downloading checkpoint from HuggingFace repo %s", hf_repo_id) logging.info("Downloading checkpoint from Hugging Face repo %s", hf_repo_id)
path = os.path.join( path = os.path.join(
huggingface_hub.snapshot_download(hf_repo_id), "model.safetensors" huggingface_hub.snapshot_download(hf_repo_id), "model.safetensors"
) )