File size: 595 Bytes
4f3439e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# 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