blitzkode / requirements-training.txt
neuralbroker's picture
Update requirements-training.txt (v2.1 production)
4f3439e verified
raw
history blame
595 Bytes
# Training dependencies — tested with Python 3.13 + CUDA 12.4 on Windows
# Install with: pip install -r requirements-training.txt
# Core ML
torch>=2.6.0
transformers>=4.56.1,<4.58
tokenizers>=0.21,<0.23
peft>=0.18.0
trl>=0.24.0
datasets>=4.0.0
accelerate>=1.0.0
safetensors>=0.4.0
# Quantisation (optional — may not build on Windows without MSVC)
# bitsandbytes>=0.49.0
# Utilities
pyarrow>=23.0.0,<24 # 24.x causes Windows access violation at import
gradio>=4.0.0
# GGUF export (bundled with llama-cpp-python, also available standalone)
# gguf
# Linting / CI
ruff>=0.4.0
mypy>=1.8.0