Skip to content
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

Relax numpy upper bound and bump deepspeed version #1990

Merged
merged 2 commits into from
Aug 29, 2024

Conversation

hvaara
Copy link
Contributor

@hvaara hvaara commented Aug 28, 2024

DeepSpeed v0.14.4 and later supports current numpy (confirmed in deepspeedai/DeepSpeed#5671).

This should allow us to get rid of the upper bound on numpy after a version bump of DeepSpeed.

See #1747 for original pin.

@hvaara
Copy link
Contributor Author

hvaara commented Aug 28, 2024

Running tests locally passed

$ make test
212 passed, 180 skipped, 592 warnings in 480.39s (0:08:00)

I'm not sure if this actually tests anything related to DeepSpeed with numpy>=2.0.0. Will the DeepSpeed integration be tested in CI?

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Copy link
Member

@lewtun lewtun left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for fixing the deps @hvaara ! LGTM with a small change

setup.py Show resolved Hide resolved
@lewtun lewtun merged commit 4f59e92 into huggingface:main Aug 29, 2024
1 check passed
@hvaara hvaara deleted the numpy-bump branch August 29, 2024 11:19
@hvaara
Copy link
Contributor Author

hvaara commented Aug 29, 2024

Thanks for the review! 🤗

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants