From 59d29856816f22e172b8d348121b24bae51b0301 Mon Sep 17 00:00:00 2001 From: Yichen Zhou Date: Thu, 11 Sep 2025 18:17:14 -0700 Subject: [PATCH] Update timesfm_2p5_torch.py Update Hugging Face repo id. --- src/timesfm/timesfm_2p5/timesfm_2p5_torch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py b/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py index c4acf05..3855d9d 100644 --- a/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py +++ b/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py @@ -264,13 +264,13 @@ class TimesFM_2p5_200M_torch(timesfm_2p5_base.TimesFM_2p5): self, *, 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.""" if path: pass 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( huggingface_hub.snapshot_download(hf_repo_id), "model.safetensors" )