huggingface / accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
https://huggingface.co/docs/accelerate
Apache License 2.0
7.76k stars 941 forks source link

Potentially fix tests #2862

Closed muellerzr closed 3 months ago

muellerzr commented 3 months ago

What does this PR do?

We've found out who didn't prepare for numpy v2.0.0: wandb :)

Jk it's deepspeed, numpy, and a few more. We'll pin for now sadly 😢

Fixes # (issue)

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.

@SunMarc

HuggingFaceDocBuilderDev commented 3 months ago

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.

muellerzr commented 3 months ago

Merging and we can see what happens. It may not fix this in the end, but we can try