requirements.txt

# ============================================================================
# Sakhi — comprehensive requirements (training + fine-tuning + runtime)
# Superset of requirements-runtime.txt. Use this if you intend to retrain or
# A/B-evaluate the Unsloth fine-tune; use requirements-runtime.txt if you only
# need to run `python api.py` (Path 1 workstation deployment).
#
# PyTorch Nightly (cu128) is pinned for RTX 5070 Ti / Blackwell SM 12.0.
# ============================================================================

--extra-index-url https://download.pytorch.org/whl/nightly/cu128
--pre
torch
torchaudio
torchvision

# === Transformers Stack (Gemma 4 support) ===
transformers>=4.51.0
accelerate>=1.5.0
sentencepiece>=0.2.0
protobuf>=5.29.0
safetensors>=0.4.0

# === Fine-Tuning (Unsloth) ===
unsloth[cu128]
peft>=0.14.0
trl>=0.15.0
datasets>=3.0.0

# === Quantization ===
bitsandbytes>=0.45.0

# === Audio Processing (training-side) ===
librosa>=0.10.1
soundfile>=0.12.1

# === Data Processing ===
pandas>=2.2.0
numpy>=1.26.0
scipy>=1.14.0

# === Synthetic Data Generation ===
openai>=1.50.0
python-dotenv>=1.0.0

# === Serving (also in requirements-runtime.txt) ===
faster-whisper>=1.0.0,<2.0.0
fastapi>=0.110
uvicorn[standard]>=0.27
python-multipart>=0.0.9
ollama>=0.4.0

# === Utilities ===
tqdm>=4.67.0
pyyaml>=6.0
huggingface-hub>=0.28.0
jsonschema>=4.23.0
rich>=13.0.0