transformers
[`Trainer`] Fix `.to` call on 4bit models
#24444
Merged

[`Trainer`] Fix `.to` call on 4bit models #24444

younesbelkada
younesbelkada fix `.to` call on 4bit models
877783c4
younesbelkada better check
b0ccdda4
younesbelkada younesbelkada requested a review from amyeroberts amyeroberts 2 years ago
HuggingFaceDocBuilderDev
amyeroberts
amyeroberts approved these changes on 2023-06-23
younesbelkada
lewtun
younesbelkada younesbelkada merged 468aed39 into main 2 years ago
younesbelkada younesbelkada deleted the fix-4bit-move branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone