初始化项目,由ModelHub XC社区提供模型
Model: Fordentinc/book-builder-bookwriter-v1 Source: Original Platform
This commit is contained in:
24
resume/requirements.txt
Normal file
24
resume/requirements.txt
Normal file
@@ -0,0 +1,24 @@
|
||||
# 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
|
||||
Reference in New Issue
Block a user