mirror of
https://gitee.com/ascend/ModelLink.git
synced 2024-12-05 13:27:53 +08:00
bf6456e04c
* requirements.txt移除apex依赖,模型训练脚本规范化加上日志存档
17 lines
195 B
Plaintext
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 |