huggingface/accelerate
在 GitHub 查看[Community Contributions] examples on distributed inference using 🤗 Accelerate
Open
#3,078 创建于 2024年9月4日
contributions-welcomegood first issuewip
描述
The inference/distributed directory houses examples on running distributed inference with accelerate:
- Phi2 for language generation
- Stable Diffusion for image generation
The strategy followed there is to load an entire model onto each GPU and sending chunks of a batch through each GPU’s model copy at a time. Synthetic data generation has become an essential toolkit for every ML Engineer. So, it'd be beneficial to extend these examples to include some more use cases:
- Image captioning
- Speech data generation
Some nice to haves:
- Include artifact serialization as done in this
- Keep the artifact serialization code under a thread to not block GPU execution
How can you help?
You could help us contribute an example on any of the above-mentioned use cases or you can come up with your own 🤗 Help us make the art of synthetic data generation scalable, easy, and accessible.