From e697c8bcc4ae9430398983bb99aaf3eea3a3b986 Mon Sep 17 00:00:00 2001
From: Dobiichi-Origami <56953648+Dobiichi-Origami@users.noreply.github.com>
Date: Thu, 14 Nov 2024 11:31:25 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=94=B1=E4=BA=8E=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E6=96=87=E6=A1=A3=E4=B8=8E=E5=AE=9E=E9=99=85=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9E=E4=B8=8D=E5=AF=B9=E5=AF=BC=E8=87=B4=E7=9A=84=20Bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 python/qianfan/model/model.py     | 2 +-
 python/qianfan/trainer/actions.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/qianfan/model/model.py b/python/qianfan/model/model.py
index 99306afe..844960ba 100644
--- a/python/qianfan/model/model.py
+++ b/python/qianfan/model/model.py
@@ -210,7 +210,7 @@ def publish(self, name: str = "", **kwargs: Any) -> "Model":
             "taskId": self.task_id,
         }
         if self.step:
-            model_version_meta["step"] = self.step
+            model_version_meta["checkpointStep"] = self.step
         model_publish_resp = ResourceModel.V2.create_custom_model(
             self.set_id,
             console_const.CreateCustomModelSourceType.Train,
diff --git a/python/qianfan/trainer/actions.py b/python/qianfan/trainer/actions.py
index c4b0853f..53e71b8f 100644
--- a/python/qianfan/trainer/actions.py
+++ b/python/qianfan/trainer/actions.py
@@ -1003,7 +1003,7 @@ def exec(
                 f"ms_{generate_letter_num_random_id(11)}",
                 model_type,
             )
-            existed_model_set_id = model_resp["result"]
+            existed_model_set_id = model_resp["result"]["modelSetId"]
 
         self.model = Model(
             task_id=self.task_id, job_id=self.job_id, set_id=existed_model_set_id