From 664fe3e867ad22dc3c7fdc89e2f5df19e5928f3c Mon Sep 17 00:00:00 2001
From: tangbinhan <tangbinhan@baidu.com>
Date: Fri, 27 Jun 2025 13:27:26 +0800
Subject: [PATCH] fix paddle from_dlpack error

Signed-off-by: tangbinhan <tangbinhan@baidu.com>
---
 fastsafetensors/frameworks/_paddle.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fastsafetensors/frameworks/_paddle.py b/fastsafetensors/frameworks/_paddle.py
index b089d00..0293d6d 100644
--- a/fastsafetensors/frameworks/_paddle.py
+++ b/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"):