From d5f55b08256633eb2ed7a6f2708568ccc9e68abb Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Fri, 10 May 2024 11:32:53 +0200 Subject: [PATCH] use snapshot_download --- src/timesfm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/timesfm.py b/src/timesfm.py index 56456f5..23635eb 100644 --- a/src/timesfm.py +++ b/src/timesfm.py @@ -24,7 +24,7 @@ import jax import jax.numpy as jnp import numpy as np import pandas as pd -from huggingface_hub import hf_hub_download +from huggingface_hub import snapshot_download from paxml import checkpoints from paxml import tasks_lib from praxis import base_hyperparams @@ -235,7 +235,7 @@ class TimesFm: step: step of the checkpoint to load. If `None`, load lastest checkpoint. """ # Download the checkpoint from Hugging Face Hub - checkpoint_path = hf_hub_download(repo_id) + checkpoint_path = snapshot_download(repo_id) # Initialize the model weights. self._logging("Constructing model weights.")