diff --git a/modules/shared.py b/modules/shared.py index 2e91f4d5..ea6c581a 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -79,7 +79,6 @@ group.add_argument('--model', type=str, help='Name of the model to load by defau group.add_argument('--lora', type=str, nargs='+', help='The list of LoRAs to load. If you want to load more than one LoRA, write the names separated by spaces.') group.add_argument('--model-dir', type=str, default='models/', help='Path to directory with all the models.') group.add_argument('--lora-dir', type=str, default='loras/', help='Path to directory with all the loras.') -group.add_argument('--model-menu', action='store_true', help='Show a model menu in the terminal when the web UI is first launched.') group.add_argument('--settings', type=str, help='Load the default interface settings from this yaml file. See settings-template.yaml for an example. If you create a file called settings.yaml, this file will be loaded by default without the need to use the --settings flag.') group.add_argument('--extensions', type=str, nargs='+', help='The list of extensions to load. If you want to load more than one extension, write the names separated by spaces.') group.add_argument('--verbose', action='store_true', help='Print the prompts to the terminal.') @@ -215,6 +214,7 @@ group.add_argument('--disable_exllama', action='store_true', help='DEPRECATED') group.add_argument('--disable_exllamav2', action='store_true', help='DEPRECATED') group.add_argument('--wbits', type=int, default=0, help='DEPRECATED') group.add_argument('--groupsize', type=int, default=-1, help='DEPRECATED') +group.add_argument('--model-menu', action='store_true', help='DEPRECATED') args = parser.parse_args() args_defaults = parser.parse_args([]) diff --git a/server.py b/server.py index 31e1c4c6..1f227350 100644 --- a/server.py +++ b/server.py @@ -218,28 +218,10 @@ if __name__ == "__main__": if extension not in shared.args.extensions: shared.args.extensions.append(extension) - available_models = utils.get_available_models() - # Model defined through --model if shared.args.model is not None: shared.model_name = shared.args.model - # Select the model from a command-line menu - elif shared.args.model_menu: - if len(available_models) == 0: - logger.error('No models are available! Please download at least one.') - sys.exit(0) - else: - print('The following models are available:\n') - for i, model in enumerate(available_models): - print(f'{i+1}. {model}') - - print(f'\nWhich one do you want to load? 1-{len(available_models)}\n') - i = int(input()) - 1 - print() - - shared.model_name = available_models[i] - # If any model has been selected, load it if shared.model_name != 'None': p = Path(shared.model_name)