mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-05 13:50:13 +06:00
afa96fffdf
* make run_generation more generic for other devices * use Accelerate to support any device type it supports. * make style * fix error usage of accelerator.prepare_model * use `PartialState` to make sure everything is running on the right device --------- Co-authored-by: statelesshz <jihuazhong1@huawei.com>
5 lines
67 B
Plaintext
5 lines
67 B
Plaintext
accelerate >= 0.21.0
|
|
sentencepiece != 0.1.92
|
|
protobuf
|
|
torch >= 1.3
|