Skip to content

Commit 63d93d1

Browse files
convert : disable uint types (#18908)
1 parent c5a3bc3 commit 63d93d1

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

convert_hf_to_gguf.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13232,17 +13232,18 @@ class LazyTorchTensor(gguf.LazyBase):
1323213232
}
1323313233

1323413234
# only used when byteswapping data. Only correct size is needed
13235+
# TODO: uncomment uint64, uint32, and uint16, ref: https://github.com/pytorch/pytorch/issues/58734
1323513236
_dtype_byteswap_map: dict[torch.dtype, type] = {
1323613237
torch.float64: np.float64,
1323713238
torch.float32: np.float32,
1323813239
torch.bfloat16: np.float16,
1323913240
torch.float16: np.float16,
1324013241
torch.int64: np.int64,
13241-
torch.uint64: np.uint64,
13242+
# torch.uint64: np.uint64,
1324213243
torch.int32: np.int32,
13243-
torch.uint32: np.uint32,
13244+
# torch.uint32: np.uint32,
1324413245
torch.int16: np.int16,
13245-
torch.uint16: np.uint16,
13246+
# torch.uint16: np.uint16,
1324613247
torch.int8: np.int8,
1324713248
torch.uint8: np.uint8,
1324813249
torch.bool: np.uint8,

0 commit comments

Comments
 (0)