shared_cmd_options.py 874 B

123456789101112131415161718
  1. import os
  2. import launch
  3. from modules import cmd_args, script_loading
  4. from modules.paths_internal import models_path, script_path, data_path, sd_configs_path, sd_default_config, sd_model_file, default_sd_model_file, extensions_dir, extensions_builtin_dir # noqa: F401
  5. parser = cmd_args.parser
  6. script_loading.preload_extensions(extensions_dir, parser, extension_list=launch.list_extensions(launch.args.ui_settings_file))
  7. script_loading.preload_extensions(extensions_builtin_dir, parser)
  8. if os.environ.get('IGNORE_CMD_ARGS_ERRORS', None) is None:
  9. cmd_opts = parser.parse_args()
  10. else:
  11. cmd_opts, _ = parser.parse_known_args()
  12. cmd_opts.webui_is_non_local = any([cmd_opts.share, cmd_opts.listen, cmd_opts.ngrok, cmd_opts.server_name])
  13. cmd_opts.disable_extension_access = cmd_opts.webui_is_non_local and not cmd_opts.enable_insecure_extension_access