diff --git a/docs/mindformers/docs/source_en/feature/start_tasks.md b/docs/mindformers/docs/source_en/feature/start_tasks.md
index fab3b044dd797a600bdaea331ebe48b8c66be941..4707a642e76321ec3e0921ca6696d3aaaa1e6538 100644
--- a/docs/mindformers/docs/source_en/feature/start_tasks.md
+++ b/docs/mindformers/docs/source_en/feature/start_tasks.md
@@ -64,7 +64,7 @@ In the root directory of the MindSpore Transformers code, execute the `run_mindf
 |     `--modal_type`     | Modal type of input data for predict. This parameter has been deprecated and will be removed in the next version. | str, optional                                                                                                                                                 | predict              |
 |     `--adapter_id`     | LoRA ID for predict. This parameter has been deprecated and will be removed in the next version.                  | str, optional                                                                                                                                                 | predict              |
 | `--predict_batch_size` | The batch size for multi-batch inference.                                                                         | int, optional                                                                                                                                                 | predict              |
-|     `--do_sample`      | Whether to use random sampling when selecting tokens for inference.                                               | int, optional, ``True`` means using sampling encoding, ``False`` means using greedy decoding.                                                                 | predict              |
+|     `--do_sample`      | Whether to use random sampling when selecting tokens for inference.                                               | bool, optional, ``True`` means using sampling encoding, ``False`` means using greedy decoding.                                                                 | predict              |
 
 ## Distributed Task Pull-up Script
 
diff --git a/docs/mindformers/docs/source_zh_cn/feature/start_tasks.md b/docs/mindformers/docs/source_zh_cn/feature/start_tasks.md
index 96e642a22906c7f4efc30f7846618ce38953ad48..22b1a7a88bc42a41504a097137349ea47bf8c23b 100644
--- a/docs/mindformers/docs/source_zh_cn/feature/start_tasks.md
+++ b/docs/mindformers/docs/source_zh_cn/feature/start_tasks.md
@@ -64,7 +64,7 @@ MindSpore Transformers提供了一键启动脚本`run_mindformer.py`和分布式
 |     `--modal_type`     | 模型推理输入对应模态。该参数已废弃，下个版本删除。 | str，可选                                              | 推理   |
 |     `--adapter_id`     | 推理的LoRA ID。该参数已废弃，下个版本删除。 | str，可选                                              | 推理   |
 | `--predict_batch_size` | 多batch推理的batch_size大小。    | int，可选                                              | 推理   |
-|     `--do_sample`      | 推理选择token时是否使用随机采样。       | int，可选，``True`` 表示使用随机采样，``False`` 代表使用贪心搜索。        | 推理   |
+|     `--do_sample`      | 推理选择token时是否使用随机采样。       | bool，可选，``True`` 表示使用随机采样，``False`` 代表使用贪心搜索。        | 推理   |
 
 ## 分布式任务拉起脚本