huggingface/accelerate

[Community Contributions] examples on distributed inference using 🀗 Accelerate

オヌプン

#3,078 opened on 2024/09/04

 (7 件のコメント) (2 件のリアクション) (0 人の担圓者)Python (626 件のフォヌク)batch import
contributions-welcomegood first issuewip

Repository metrics

Stars
 (5,805 個のスタヌ)
PR merge metrics
 (平均マヌゞ 16d 17h) (30d で 23 merged PRs)

説明

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.

コントリビュヌタヌガむド