diff --git a/kashgari/tasks/abs_task_model.py b/kashgari/tasks/abs_task_model.py
index b40c9bc4..6da8ea99 100644
--- a/kashgari/tasks/abs_task_model.py
+++ b/kashgari/tasks/abs_task_model.py
@@ -76,13 +76,7 @@ def default_hyper_parameters(cls) -> Dict[str, Dict[str, Any]]:
         """
         raise NotImplementedError
 
-    def save(self, model_path: str, h5_weight: bool = False) -> str:
-        """
-        Save model
-        Args:
-            model_path: target model path
-            h5_weight: whether using original h5 format or new saved_model format
-        """
+    def save(self, model_path: str) -> str:
         pathlib.Path(model_path).mkdir(exist_ok=True, parents=True)
         model_path = os.path.abspath(model_path)