Upload requirements.txt

requirements.txt

@@ -1,52 +1,5 @@
-# PolyBloom ML Service v19.1.0 — Hugging Face Space
-# Python 3.11 (HF default).
-#
-# FIX LOG:
-#   v1: uni2ts<1.0 matched nothing
-#   v2: uni2ts>=1.1 conflicts with scipy>=1.12 (uni2ts pins scipy==1.11.3) → DROP uni2ts
-#   v3: momentfm pins numpy==1.25.2 + huggingface-hub==0.19.4, conflicts with our >=1.26/>=0.22 → DROP momentfm
-#   v4: added neuralforecast>=1.7.0 (covers PatchTST, N-HiTS, iTransformer)
-#       neuralforecast pins torch>=2.1 and uses optuna/ray — compatible with our stack
-#       chronos switched to PyPI package (chronos-forecasting) for HF compatibility
-#
-# Both MOIRAI and MOMENT fall back to bootstrap formulas on the free HF tier anyway (OOM).
-# Neither uni2ts nor momentfm provide any runtime benefit on a free CPU space.
-# To re-enable on a paid GPU space:
-#   MOIRAI  → add uni2ts>=1.1,<2.0  + pin scipy==1.11.3
-#   MOMENT  → add momentfm>=0.1     + pin numpy==1.25.2 + huggingface-hub==0.19.4
-
-# ── API framework ─────────────────────────────────────────────────────────────
-fastapi==0.115.*
-uvicorn[standard]==0.30.*
-
-# ── Numerics ──────────────────────────────────────────────────────────────────
-numpy>=1.26,<2.0
-scipy>=1.12,<2.0
-pandas>=2.0,<3.0
-
-# ── Trained classifiers (Row 2 — always bootstrap on startup, always green) ──
-xgboost>=2.0,<3.0
-lightgbm>=4.3,<5.0
-catboost>=1.2,<2.0
-
-# ── Chronos (Amazon T5 zero-shot) — Row 1 foundation (fallback if OOM) ───────
-# PyPI package preferred over git+ on HF Spaces (faster cold start, cached wheels)
-chronos-forecasting>=1.3,<2.0
-
-# ── MOIRAI  → uni2ts DROPPED (scipy conflict) — bootstrap fallback active ────
-# ── MOMENT  → momentfm DROPPED (numpy + hf-hub conflict) — bootstrap active ──
-
-# ── NeuralForecast — Row 3 (PatchTST, N-HiTS, iTransformer) ─────────────────
-# Single package covers all three models; bootstrap-trains on startup (CPU safe).
-# Requires torch>=2.1 (satisfied below). Pulls in lightning, optuna automatically.
-neuralforecast>=1.7.0,<2.0
-
-# ── Shared transformer / neural stack ────────────────────────────────────────
-torch>=2.2,<3.0
-transformers>=4.40,<5.0
-huggingface-hub>=0.22,<1.0
-einops>=0.7,<1.0
-
-# ── Utilities ─────────────────────────────────────────────────────────────────
-scikit-learn>=1.4,<2.0
-python-multipart>=0.0.9
+fastapi==0.111.0
+uvicorn[standard]==0.29.0
+numpy==1.26.4
+torch==2.2.2
+git+https://github.com/amazon-science/chronos-forecasting.git