diff --git a/README.md b/README.md
index 700a7eb3071795d513fa47b0e7da567ad1de568e..453453e006a92af688e729eed20da290ac957a1b 100644
--- a/README.md
+++ b/README.md
@@ -113,7 +113,7 @@ paddlepalm框架的运行原理图如图所示
 我们提供了BERT、ERNIE等主干网络的相关预训练模型。为了加速模型收敛，获得更佳的测试集表现，我们强烈建议用户在多任务学习时尽量在预训练模型的基础上进行（而不是从参数随机初始化开始）。用户可以查看可供下载的预训练模型：
 
 ```shell
-python download_models.py
+python download_models.py ls pretrain
 ```
 
 用户可通过运行`python download_models.py download <model_name>`下载需要的预训练模型，例如，下载预训练BERT模型（uncased large）的命令如下：
diff --git a/download_models.py b/download_models.py
index 620e84d266bd8505d96e8754a3547705f66fa98e..4c7314e2c3ddd3af5abcc58cbdc4284314c6b45e 100644
--- a/download_models.py
+++ b/download_models.py
@@ -16,6 +16,6 @@
 import paddlepalm as palm
 import sys
 if(sys.argv[1] == 'ls'):
-  palm.ls(sys.argv[1], sys.argv[2])
+  palm.ls(sys.argv[2])
 if(sys.argv[1] == 'download'):
-  palm.download(sys.argv[1], sys.argv[2])
+  palm.download('pretrain', sys.argv[2])