transformers
[QoL] Allow dtype str for torch_dtype arg of from_pretrained
#31590
Merged
Overview
Commits
3
Changes
View On GitHub

[QoL] Allow dtype str for torch_dtype arg of from_pretrained #31590

ArthurZucker merged 3 commits into huggingface:main from aliencaocao:allow-string-torch-dtype
aliencaocao
aliencaocao Allow dtype str for torch_dtype in from_pretrained
98d274d2
aliencaocao Update docstring
04740875
aliencaocao aliencaocao changed the title Allow dtype str for torch_dtype in from_pretrained [QoL] Allow dtype str for torch_dtype in from_pretrained 1 year ago
aliencaocao aliencaocao changed the title [QoL] Allow dtype str for torch_dtype in from_pretrained [QoL] Allow dtype str for torch_dtype arg of from_pretrained 1 year ago
amyeroberts
amyeroberts commented on 2024-06-25
aliencaocao Add tests for str torch_dtype
6b30a34e
aliencaocao
amyeroberts amyeroberts requested a review from ArthurZucker ArthurZucker 1 year ago
amyeroberts
amyeroberts approved these changes on 2024-06-26
ArthurZucker
ArthurZucker approved these changes on 2024-06-27
ArthurZucker ArthurZucker merged 3a028101 into main 1 year ago
HuggingFaceDocBuilderDev
aliencaocao aliencaocao deleted the allow-string-torch-dtype branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum