transformers
3a028101 - [QoL] Allow dtype str for torch_dtype arg of from_pretrained (#31590)

Commit
1 year ago
[QoL] Allow dtype str for torch_dtype arg of from_pretrained (#31590) * Allow dtype str for torch_dtype in from_pretrained * Update docstring * Add tests for str torch_dtype
Author
Parents
Loading