Update `dtype_byte_size` to handle torch.float8_e4m3fn/float8_e5m2 types #30488
Update modeling_utils/dtype_byte_size to handle float8 types
9119b7bd
mgoin
changed the title Update `modeling_utils/dtype_byte_size` to handle float8 types Update `dtype_byte_size` to handle torch.float8_e4m3fn/float8_e5m2 types 2 years ago
Add a test for dtype_byte_size
81ce419c
asartran
approved these changes
on 2024-04-25
Format
db147751
Fix bool
a0eb75c0
asartran
merged
20081c74
into main 2 years ago
mgoin
deleted the patch-1 branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub