diff --git a/.github/workflows/test-invoke-pip.yml b/.github/workflows/test-invoke-pip.yml index 34c90f10c5..4985202782 100644 --- a/.github/workflows/test-invoke-pip.yml +++ b/.github/workflows/test-invoke-pip.yml @@ -119,6 +119,7 @@ jobs: run: > configure_invokeai.py --yes + --default_only --full-precision # can't use fp16 weights without a GPU - name: Run the tests diff --git a/docs/installation/020_INSTALL_MANUAL.md b/docs/installation/020_INSTALL_MANUAL.md index c1181db347..ead1eb6536 100644 --- a/docs/installation/020_INSTALL_MANUAL.md +++ b/docs/installation/020_INSTALL_MANUAL.md @@ -217,7 +217,7 @@ manager, please follow these steps: !!! tip - Do not move the source code repository after installation. The virtual environment directory has absolute paths in it that get confused if the directory is moved. + Do not move the source code repository after installation. The virtual environment directory has absolute paths in it that get confused if the directory is moved. --- diff --git a/scripts/configure_invokeai.py b/scripts/configure_invokeai.py index 22ade19cdb..9d17a73317 100755 --- a/scripts/configure_invokeai.py +++ b/scripts/configure_invokeai.py @@ -197,6 +197,14 @@ def recommended_datasets()->dict: datasets[ds]=True return datasets +#--------------------------------------------- +def default_dataset()->dict: + datasets = dict() + for ds in Datasets.keys(): + if Datasets[ds].get('default',False): + datasets[ds]=True + return datasets + #--------------------------------------------- def all_datasets()->dict: datasets = dict() @@ -646,7 +654,7 @@ def download_weights(opt:dict) -> Union[str, None]: precision = 'float32' if opt.full_precision else choose_precision(torch.device(choose_torch_device())) if opt.yes_to_all: - models = recommended_datasets() + models = default_dataset() if opt.default_only else recommended_datasets() access_token = authenticate(opt.yes_to_all) if len(models)>0: successfully_downloaded = download_weight_datasets(models, access_token, precision=precision) @@ -808,6 +816,9 @@ def main(): dest='yes_to_all', action='store_true', help='answer "yes" to all prompts') + parser.add_argument('--default_only', + action='store_true', + help='when --yes specified, only install the default model') parser.add_argument('--config_file', '-c', dest='config_file',