No module named 'transformers.deepspeed'

microsoft / DeepSpeedExamples

Example models using DeepSpeed

Apache License 2.0

6.1k stars 1.04k forks source link

No module named 'transformers.deepspeed' #934

Closed TianyuJIAA closed 3 weeks ago

TianyuJIAA commented 3 weeks ago

when i run the script "python e2e_rlhf.py --actor-model facebook/opt-1.3b --reward-model facebook/opt-350m --deployment-type single_gpu", it raise "ModuleNotFoundError: No module named 'transformers.deepspeed'".

I am very certain that all required Python libraries listed in the requirements file were installed before running the script.

error info:

python package:

note that python version == 3.12.3

zhulu-bot commented 3 weeks ago

pip install transformers==4.37.2 I solved the problem by changing the version

loadams commented 3 weeks ago

Hi @TianyuJIAA - We've merged the PR that resolves this here: https://github.com/microsoft/DeepSpeedExamples/pull/872

Please try again with the latest DeepSpeedExamples code.

Please comment if you need any more help and we can re-open the issue.