Open davidmrau opened 3 months ago
Add multi-gpu support with accelerate. Requires wrapping the model as well as data with the accelerate function.
Any update?
Add multi-gpu support with accelerate. Requires wrapping the model as well as data with the accelerate function.