New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined #31751

Merged

LysandreJik merged 1 commit into huggingface:main from kiszk:fix_device_with_accelarate

Jul 5, 2024

Merged

Make tensor device correct when ACCELERATE_TORCH_DEVICE is defined #31751

return correct device when ACCELERATE_TORCH_DEVICE is defined

CircleCI Checks / setup_and_quality succeeded Jul 2, 2024 in 3m 54s

Workflow: setup_and_quality

View CircleCI workflow

View more details on CircleCI Checks