diff --git a/hpvm/scripts/hpvm_installer.py b/hpvm/scripts/hpvm_installer.py index e83d5f3e727044dcba942e2533597bc779c9b816..f99a333a60a4c4525941a3facd0d5d1ad8a8496e 100755 --- a/hpvm/scripts/hpvm_installer.py +++ b/hpvm/scripts/hpvm_installer.py @@ -120,11 +120,15 @@ def prompt_args(): "Build target: ", parse_targets, "Input shouldn't contain space" ) + download_weights = input_with_check( + "Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n" + ) run_tests = input_with_check( "Build and run tests? [y/n]: ", parse_yn, "Please enter y or n" ) return Namespace( - no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests + no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests, + no_params=not download_weights )