--- license: apache-2.0 library_name: pytorch pipeline_tag: time-series-forecasting tags: - time-series - time-series-forecasting - zero-shot - probabilistic-forecasting - vision --- # ViTime Official pretrained checkpoint for **ViTime: Foundation Model for Time Series Forecasting Powered by Vision Intelligence**. - Code: https://github.com/IkeYang/ViTime - Paper: https://openreview.net/forum?id=XInsJDBIkp ViTime converts numerical time series into binary images and performs point and probabilistic forecasting with a vision-based architecture. ## Checkpoint | File | Size | SHA-256 | | --- | ---: | --- | | `ViTime_Model.pth` | 296,778,986 bytes | `6513b03b352163337c333526fd3634b07db789665cd9f87648f9661efe0cff1a` | The stable release revision is `v1.0.0`. ## Download Public downloads do not require a Hugging Face account or access token. ```python from huggingface_hub import hf_hub_download checkpoint_path = hf_hub_download( repo_id="IkeYEUNG/ViTime", filename="ViTime_Model.pth", revision="v1.0.0", ) print(checkpoint_path) ``` Command-line download: ```bash hf download IkeYEUNG/ViTime ViTime_Model.pth --revision v1.0.0 ``` ## Use with the official GitHub repository ```bash git clone https://github.com/IkeYang/ViTime.git cd ViTime python -m pip install -r requirements.txt ``` The official code downloads this `v1.0.0` checkpoint automatically on first use and reuses the Hugging Face cache on later runs: ```python import numpy as np from main import ViTimePrediction x = np.sin(np.arange(512) / 10) model = ViTimePrediction(device="cuda:0", model_name="MAE", lookbackRatio=None) prediction = model.prediction(x, future_length=720) print(prediction.shape) ``` See the GitHub repository for installation requirements and complete point/probabilistic forecasting examples. ## Checkpoint security This release preserves the original PyTorch `.pth` checkpoint format for compatibility with the official code. Load serialized PyTorch checkpoints only from trusted sources and pin the `v1.0.0` revision for reproducible use. ## Citation ```bibtex @article{yang2025vitime, title={{ViTime}: Foundation Model for Time Series Forecasting Powered by Vision Intelligence}, author={Yang, Luoxiao and Wang, Yun and Fan, Xinqi and Cohen, Israel and Chen, Jingdong and Zhang, Zijun}, journal={Transactions on Machine Learning Research}, year={2025}, url={https://openreview.net/forum?id=XInsJDBIkp}, note={Published in Transactions on Machine Learning Research (10/2025)} } ```