diff --git a/download_models.py b/download_models.py index 4c7314e2c3ddd3af5abcc58cbdc4284314c6b45e..7095a916ab889754dc983bb7e8305c17cb31ca12 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[2]) + palm.downloader.ls(sys.argv[2]) if(sys.argv[1] == 'download'): - palm.download('pretrain', sys.argv[2]) + palm.downloader.download('pretrain', sys.argv[2])