fix paddle from_dlpack error

Author: tangbinhan

fastsafetensors/frameworks/_paddle.py

@@ -21,7 +21,7 @@
 
 dtype_convert: Dict[DType, Any] = {
     DType.BOOL: paddle.bool,
-    DType.I8: paddle.uint8,
+    DType.U8: paddle.uint8,
     DType.I8: paddle.int8,
     DType.I16: paddle.int16,
     DType.U16: paddle.bfloat16,
@@ -37,6 +37,7 @@
 need_workaround_dtypes: Dict[DType, DType] = {
     DType.F8_E5M2: DType.I8,
     DType.F8_E4M3: DType.I8,
+    DType.U16: DType.BF16,
 }
 
 if hasattr(paddle, "float8_e5m2"):
@@ -198,7 +199,7 @@ def concat_tensors(self, tensors: List[PaddleTensor], dim) -> PaddleTensor:
     def get_dtype_size(self, dtype: DType) -> int:
         return paddle_core.size_of_dtype(dtype_convert[dtype])
 
-    def from_dlpack(self, dl_tensor: Any, device: Device, dtype: DType) -> PaddleTensor:
+    def from_dlpack(self, dl_tensor: Any, device: Device, dtype) -> PaddleTensor:
         return PaddleTensor(device, dtype, paddle.utils.dlpack.from_dlpack(dl_tensor))
 
     def copy_tensor(self, dst: PaddleTensor, src: PaddleTensor) -> None: