-
Notifications
You must be signed in to change notification settings - Fork 27k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Install accelerete@main
in PyTorch Past CI jobs.
#22963
Conversation
if: inputs.framework == 'pytorch' | ||
working-directory: /transformers | ||
run: | | ||
python3 -m pip install --no-cache-dir git+https://github.com/huggingface/accelerate@main#egg=accelerate |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
need this to be installed in CI jobs, not in Past CI docker images. See PR description.
@@ -39,6 +38,10 @@ RUN $INSTALL_CMD | |||
|
|||
RUN [ "$FRAMEWORK" != "pytorch" ] && echo "`deepspeed-testing` installation is skipped" || python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing] | |||
|
|||
# Remove `accelerate`: it requires `torch`, and this causes import issues for TF-only testing | |||
# We will install `accelerate@main` in Past CI workflow file | |||
RUN python3 -m pip uninstall -y accelerate |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Installed in Past CI PyTorch runs.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks!
The documentation is not available anymore as the PR was closed or merged. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Makes sense!
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
What does this PR do?
Install
accelerete@main
in PyTorch Past CI jobs.Context
In ##22393, we added back
deepspeed
in Past CI docker image. Later in #22859, we decide to useaccelerate@main
, but I forgot to apply the same change in Past CI docker file, as I mistakenly thought Past CI doesn't useaccelerate
.[deepspeed-testing]
(installed in Past CI docker) includesaccelerate
, and we want it to beaccelerate@main
.acclerate
it in the docker image, as for the TF Past CI, it will break somethingaccelerate requires torch, and this causes import issues for TF-only testing