transformers
73fdc8c5 - [deepspeed zero3] need `generate(synced_gpus=True, ...)` (#22242)

Commit
2 years ago
[deepspeed zero3] need `generate(synced_gpus=True, ...)` (#22242) * [deepspeed zero3] need generate(synced_gpus=True, ...) * fix * rework per Sylvain's suggestion * Apply suggestions from code review Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> --------- Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Author
Parents
Loading