accelerate
Only check for dtype if it has it in get_state_dict
#1288
Merged

Only check for dtype if it has it in get_state_dict #1288

muellerzr merged 1 commit into main from state-dict
muellerzr
muellerzr Check for dtype attr
9242310a
muellerzr muellerzr added enhancement
muellerzr muellerzr requested a review from sgugger sgugger 3 years ago
sgugger
sgugger approved these changes on 2023-04-03
HuggingFaceDocBuilderDev
muellerzr muellerzr merged 07ad358f into main 3 years ago
muellerzr muellerzr deleted the state-dict branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone