tests/test_modeling_utils.py · 884e3b1c53099c8d88b3897b903eb79f7cc37c51 · chenpangpang / transformers · GitLab

"tests/models/rembert/__init__.py" did not exist on "31c23bd5ee26425a67f92fc170789656379252a6"

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