requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121

# Core compute (CUDA 12.x -> cu121 wheels)
torch==2.3.1+cu121

# HF stack (stable pair)
transformers==4.44.2
tokenizers==0.19.1          # compatible with 4.44.2
sentencepiece==0.1.99
huggingface_hub==0.24.6
accelerate==0.33.0
safetensors==0.4.4
bitsandbytes==0.43.1
regex==2024.7.24

# UI
gradio==4.44.0
gradio_client==1.3.0
websockets>=12.0

# Data helpers
pandas==2.2.2
numpy==1.26.4