tests/test_modeling_utils.py · 20081c743ee2ce31d178f2182c7466c3313adcd2 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Update `dtype_byte_size` to handle torch.float8_e4m3fn/float8_e5m2 types (#30488) · 20081c74
Michael Goin authored Apr 26, 2024
```
* Update modeling_utils/dtype_byte_size to handle float8 types

* Add a test for dtype_byte_size

* Format

* Fix bool
```
20081c74

test_modeling_utils.py 103 KB