fbmc-chronos2 / requirements.txt
Evgueni Poloukarov
fix: revert PyTorch + redirect cache to /tmp for A100
73e9c10
# Gradio
gradio==4.44.0
# Core ML/Data
torch>=2.0.0
transformers>=4.35.0
chronos-forecasting>=1.2.0
datasets>=2.14.0
polars>=0.19.0
pyarrow>=13.0.0
# HuggingFace
huggingface-hub>=0.19.0
# Visualization (for local analysis)
altair>=5.0.0
# Utilities
python-dotenv
tqdm
# Cache bust: v1.7.0 - Revert PyTorch + cache to /tmp for A100