transformers
Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined
#31751
Merged
Merged
Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined #31751
LysandreJik
merged
1 commit into
huggingface:main
from
kiszk:fix_device_with_accelarate
return correct device when ACCELERATE_TORCH_DEVICE is defined
b1c1c93e
amyeroberts
approved these changes
on 2024-07-02
muellerzr
approved these changes
on 2024-07-03
LysandreJik
approved these changes
on 2024-07-05
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub