# 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