accelerate
18121523 - Add log message for RTX 4000 series when performing multi-gpu inference with device_map (#2557)

Commit
1 year ago
Add log message for RTX 4000 series when performing multi-gpu inference with device_map (#2557) * add log message for RTX 4000 series when using device_map multi-gpu * style * style * switch to warning * Update src/accelerate/big_modeling.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> --------- Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Author
Parents
Loading