25 lines
698 B
Plaintext
25 lines
698 B
Plaintext
|
# Exact pinned versions for resuming book-builder-bookwriter-v1 training.
|
||
|
|
# These versions were verified working end-to-end on B200 (sm_100) and 4090 (sm_89).
|
||
|
|
# Install with:
|
||
|
|
# pip install -r requirements.txt --index-url https://download.pytorch.org/whl/cu128
|
||
|
|
# (only torch needs cu128 wheel; rest install from PyPI normally — see install commands below)
|
||
|
|
|
||
|
|
# === Core (install via cu128 PyTorch index) ===
|
||
|
|
torch==2.12.0
|
||
|
|
triton==3.7.0
|
||
|
|
|
||
|
|
# === HF stack ===
|
||
|
|
transformers==4.46.3
|
||
|
|
accelerate==1.1.1
|
||
|
|
datasets==3.1.0
|
||
|
|
peft==0.13.2
|
||
|
|
bitsandbytes==0.49.2
|
||
|
|
huggingface-hub==0.36.2
|
||
|
|
hf-transfer==0.1.9
|
||
|
|
safetensors==0.7.0
|
||
|
|
sentencepiece==0.2.1
|
||
|
|
protobuf==7.35.0
|
||
|
|
tokenizers==0.20.3
|
||
|
|
tensorboard==2.20.0
|
||
|
|
numpy==2.4.6
|