Michael Goin 20081c743e Update dtype_byte_size to handle torch.float8_e4m3fn/float8_e5m2 types (#30488) ...

* Update modeling_utils/dtype_byte_size to handle float8 types

* Add a test for dtype_byte_size

* Format

* Fix bool

2024-04-26 11:26:43 +01:00

103 KiB

Executable File

Raw Blame History

View Raw