accelerate
Set correct NPU backend and distributed_type when using transfer_to_npu
#3021

Merged

Set correct NPU backend and distributed_type when using transfer_to_npu #3021

muellerzr merged 3 commits into huggingface:main from ArthurinRUC:npu-set-device

fix npu setting

f699b4b2

fix npu setting

f1917d4b

muellerzr commented on 2024-08-16

add code comments

9edbc7b3

muellerzr approved these changes on 2024-08-17

BenjaminBossan approved these changes on 2024-08-19

muellerzr merged 5536a3a8 into main 1 year ago

Reviewers

BenjaminBossan

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone