SalesforceAIResearch / uni2ts

[ICML2024] Unified Training of Universal Time Series Forecasting Transformers
Apache License 2.0
612 stars 49 forks source link

Update huggingface model weights to use PyTorchModelHubMixin #39

Closed gorold closed 1 month ago

gorold commented 1 month ago

This pull request updates the codebase to download pre-trained model weights via the PyTorchModelHubMixin, which supports the tracking of download metrics. It also contains some minor changes to moirai_forecast_pandas.ipynb to fix formatting issues and also use plain pytorch for the first example.