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

Replace ModelHubMixin with PytorchModelHubMixin

Browse Source
This commit is contained in:
Kashif Rasul
2026-03-11 09:36:14 +01:00
committed by GitHub
parent a8bcd5d6d9
commit 0134e55a6e
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
@@ -22,7 +22,7 @@ from typing import Optional, Sequence, Union
import numpy as np import numpy as np
import torch 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 safetensors.torch import load_file, save_file
from torch import nn from torch import nn
@@ -265,7 +265,7 @@ class TimesFM_2p5_200M_torch_module(nn.Module):
class TimesFM_2p5_200M_torch( class TimesFM_2p5_200M_torch(
timesfm_2p5_base.TimesFM_2p5, timesfm_2p5_base.TimesFM_2p5,
ModelHubMixin, PytorchModelHubMixin,
library_name="timesfm", library_name="timesfm",
repo_url="https://github.com/google-research/timesfm", repo_url="https://github.com/google-research/timesfm",
paper_url="https://arxiv.org/abs/2310.10688", paper_url="https://arxiv.org/abs/2310.10688",