From fea4cf11eeb8db4ed080b8d3851011037675008d Mon Sep 17 00:00:00 2001
From: liufengwei0103 <2472937968@qq.com>
Date: Fri, 7 Nov 2025 21:12:17 +0800
Subject: [PATCH] save_opt

---
 python/paddle/framework/io.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/framework/io.py b/python/paddle/framework/io.py
index d56b64c230c797..56ec7900465968 100644
--- a/python/paddle/framework/io.py
+++ b/python/paddle/framework/io.py
@@ -178,7 +178,7 @@ def _build_saved_state_dict(state_dict):
                     and core.is_compiled_with_custom_device('npu')
                 ):
                     value = paddle._C_ops.npu_identity(value, -1)
-                save_dict[key] = np.array(value.cpu())
+                save_dict[key] = value.numpy()
             name_table[key] = value.name
         else:
             save_dict[key] = value