typo

10 months ago · 6195f7e6c4
parent 9abef5a66c
commit 6195f7e6c4
1 changed files with 1 additions and 1 deletions
--- a/src/petals/utils/misc.py
+++ b/src/petals/utils/misc.py
@ -9,7 +9,7 @@ def is_dummy(tensor: torch.Tensor):
    return tensor.numel() == 0


-SPECIAL_DTYPE_SIZES = {torch.bool: 1, torch.int8: 1, torch.qint32: 4}
+SPECIAL_DTYPE_SIZES = {torch.bool: 1, torch.qint8: 1, torch.qint32: 4}


 def get_size_in_bytes(dtype: torch.dtype) -> int: