We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e32703f commit 664fe3eCopy full SHA for 664fe3e
fastsafetensors/frameworks/_paddle.py
@@ -21,7 +21,7 @@
21
22
dtype_convert: Dict[DType, Any] = {
23
DType.BOOL: paddle.bool,
24
- DType.I8: paddle.uint8,
+ DType.U8: paddle.uint8,
25
DType.I8: paddle.int8,
26
DType.I16: paddle.int16,
27
DType.U16: paddle.bfloat16,
@@ -37,6 +37,7 @@
37
need_workaround_dtypes: Dict[DType, DType] = {
38
DType.F8_E5M2: DType.I8,
39
DType.F8_E4M3: DType.I8,
40
+ DType.U16: DType.BF16,
41
}
42
43
if hasattr(paddle, "float8_e5m2"):
0 commit comments