模型压缩任务参数 config 更名为 comp_config

baidubce · Dec 3, 2024 · ce6ac05 · ce6ac05
1 parent f699e5e
commit ce6ac05
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/qianfan/model/model.py b/python/qianfan/model/model.py
@@ -344,7 +344,7 @@ def compress(
         model_comp_task_resp = ResourceModel.V2.create_model_comp_task(
             name=f"mco_{generate_letter_num_random_id(12)}",
             source_model_id=self.id,
-            config=config,
+            comp_config=config,
             model_set_id=self.set_id,
             description=f"mcomp_{self.id}_{strategy.value}_{weight}",
         )

diff --git a/python/qianfan/resources/console/model.py b/python/qianfan/resources/console/model.py
@@ -1152,7 +1152,7 @@ def create_model_comp_task(
             cls,
             name: str,
             source_model_id: str,
-            config: Dict[str, Any],
+            comp_config: Dict[str, Any],
             model_set_id: str,
             description: Optional[str] = None,
         ) -> QfRequest:
@@ -1164,7 +1164,7 @@ def create_model_comp_task(
                 compression task name
             source_model_id: str,
                 source_model version id, e.g. amv-xxxx
-            config: dict,
+            comp_config: dict,
                 configuration for compressing
             model_set_id: str,
                 target model set id, e.g. am-xxx
@@ -1188,7 +1188,7 @@ def create_model_comp_task(
                     "name": name,
                     "sourceModelId": source_model_id,
                     "modelSetId": model_set_id,
-                    "config": config,
+                    "config": comp_config,
                 },
             )
             if description is not None: