transformers
[run_clm example] add torch_dtype option for model load.
#20971
Merged

[run_clm example] add torch_dtype option for model load. #20971

sywangyi
sywangyi [run_clm example] add torch_dtype option for model load.
aaf2d24a
sywangyi
HuggingFaceDocBuilderDev
sgugger
sgugger commented on 2023-01-03
sywangyi add other type in option
3e356eb0
sywangyi
sgugger
sgugger approved these changes on 2023-01-03
sywangyi fix style
607baf20
sgugger sgugger merged 9c9fe89f into main 3 years ago
sywangyi sywangyi deleted the add_torch_dtype branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone