intelligent-machine-learning / dlrover

DLRover: An Automatic Distributed Deep Learning System
Other
1.27k stars 167 forks source link

transformers version? #1221

Closed Alex-Ruan closed 3 months ago

Alex-Ruan commented 3 months ago

tranformers version I installed:

pip install transformers==4.43.3

Problem

ImportError: cannot import name 'unwrap_model' from 'transformers.trainer' (/opt/conda/lib/python3.10/site-packages/transformers/trainer.py)

Question

Which version should I install? Is there a requirement.txt?

Alex-Ruan commented 3 months ago

I find: dlrover-master\examples\pytorch\llama2\requirements.txt