PR #1506 Fix 4bit model on multiple devices

Fix 4bit model on multiple devices #1506

SunMarc merged 2 commits into huggingface:main from SunMarc:training_device_map

Add 4bit case and fix device index

399f8df0

SunMarc requested a review from

younesbelkada 2 years ago

Fix style

d80b3acd

SunMarc requested a review from

sgugger 2 years ago

sgugger approved these changes on 2023-06-01

younesbelkada approved these changes on 2023-06-01

SunMarc merged d4262021 into main 2 years ago

SunMarc deleted the training_device_map branch 2 years ago

Reviewers

sgugger

younesbelkada

Assignees

No one assigned

Labels

None yet

Milestone

No milestone