15 lines
No EOL
332 B
Text
15 lines
No EOL
332 B
Text
# CPU version of PyTorch
|
|
torch>=2.0.0 --index-url https://download.pytorch.org/whl/cpu
|
|
transformers>=4.40.0
|
|
accelerate>=0.27.0
|
|
peft>=0.11.0
|
|
trl>=0.9.0
|
|
datasets>=2.18.0
|
|
bitsandbytes>=0.43.0
|
|
wandb>=0.16.0
|
|
pyyaml>=6.0
|
|
jsonlines>=4.0.0
|
|
scikit-learn>=1.3.0
|
|
# flash-attn is not needed for CPU version
|
|
sentencepiece>=0.2.0
|
|
protobuf>=4.25.0 |