diff --git a/scripts/configure_invokeai.py b/scripts/configure_invokeai.py old mode 100644 new mode 100755 index 2bfefaa28c..2f9b6c4fbc --- a/scripts/configure_invokeai.py +++ b/scripts/configure_invokeai.py @@ -108,11 +108,13 @@ completely skip this step. completer.complete_extensions(None) # turn off path-completion mode selection = None while selection is None: - choice = input('Download ecommended models, ustomize the list, or kip this step? [r]: ') + choice = input('Download ecommended models, ll models, ustomized list, or kip this step? [r]: ') if choice.startswith(('r','R')) or len(choice)==0: selection = 'recommended' elif choice.startswith(('c','C')): selection = 'customized' + elif choice.startswith(('a','A')): + selection = 'all' elif choice.startswith(('s','S')): selection = 'skip' return selection @@ -166,7 +168,14 @@ def recommended_datasets()->dict: if Datasets[ds]['recommended']: datasets[ds]=True return datasets - + +#--------------------------------------------- +def all_datasets()->dict: + datasets = dict() + for ds in Datasets.keys(): + datasets[ds]=True + return datasets + #-------------------------------Authenticate against Hugging Face def authenticate(): print(''' @@ -526,6 +535,8 @@ def download_weights(opt:dict): if choice == 'recommended': models = recommended_datasets() + elif choice == 'all': + models = all_datasets() elif choice == 'customized': models = select_datasets(choice) if models is None and yes_or_no('Quit?',default_yes=False):