From 0134e55a6ec207d6e4c4636b12b0dd4f4c210584 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 11 Mar 2026 09:36:14 +0100 Subject: [PATCH] Replace ModelHubMixin with PytorchModelHubMixin --- 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 bd9d37a..e438abb 100644 --- a/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py +++ b/src/timesfm/timesfm_2p5/timesfm_2p5_torch.py @@ -22,7 +22,7 @@ from typing import Optional, Sequence, Union import numpy as np import torch -from huggingface_hub import ModelHubMixin, hf_hub_download +from huggingface_hub import PytorchModelHubMixin, hf_hub_download from safetensors.torch import load_file, save_file from torch import nn @@ -265,7 +265,7 @@ class TimesFM_2p5_200M_torch_module(nn.Module): class TimesFM_2p5_200M_torch( timesfm_2p5_base.TimesFM_2p5, - ModelHubMixin, + PytorchModelHubMixin, library_name="timesfm", repo_url="https://github.com/google-research/timesfm", paper_url="https://arxiv.org/abs/2310.10688",