ModelLink2/requirements.txt
LeiZhenzhen bf6456e04c !1074 requirements.txt移除apex依赖,模型训练脚本规范化加上日志存档
* requirements.txt移除apex依赖,模型训练脚本规范化加上日志存档
2024-03-19 10:55:11 +00:00

17 lines
195 B
Plaintext

numpy
transformers==4.36.0
transformers_stream_generator
sympy
decorator
scipy
sentencepiece
einops
datasets
pybind11
accelerate
six
torch==2.1.0
torchvision==0.16.0
protobuf
peft==0.7.1
tiktoken