ModelMatrix / requirements.txt
Akshay4506's picture
Fix single-row prediction error and enable high-speed HF transfers
73d00fd
raw
history blame contribute delete
748 Bytes
# Core Backend & API
fastapi>=0.110.0
uvicorn[standard]>=0.29.0
python-multipart>=0.0.9
python-dotenv>=1.0.0
jinja2>=3.1.3
hf-transfer
hf_xet
# Data & Science Stack
numpy==1.26.4
pandas==2.2.3
scikit-learn==1.6.1
scipy==1.14.1
matplotlib==3.9.2
seaborn==0.13.2
pyyaml==6.0.2
tqdm==4.67.1
joblib==1.4.2
psutil==6.1.1
openml==0.14.2
# Machine Learning Models
xgboost>=2.0.0
lightgbm>=4.0.0
catboost>=1.2.0
tabpfn==0.1.11
# PyTorch & Hugging Face (for SAP RPT-1 OSS)
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.7.0+cpu
transformers==4.52.4
accelerate==1.6.0
huggingface-hub==0.30.2
datasets==3.5.0
pyarrow==20.0.0
torcheval==0.0.7
# SAP RPT-1 OSS model
sap-rpt-oss @ git+https://github.com/SAP-samples/sap-rpt-1-oss.git@v1.1.2