Fix run as root flag (bef36597) · Commits · github_fork / Stable Diffusion Webui

Even though -f enables running webui.sh as root, the -f flag will also be passed to launch.py, causing it to exit with a usage message. This adds a line to launch.py to remove the -f flag if present. In addition to the above, all the letters in the command line arguments after each '-' were being processed for 'f' and "illegal option" was displayed for each letter that didn't match. Instead, this commit silences those errors and stops processing if the first flag doesn't start with '-f'.

launch.py

+1 −0

Original line number	Diff line number	Diff line
		@@ -186,6 +186,7 @@ def prepare_enviroment():
		parser.add_argument("--ui-settings-file", type=str, help="filename to use for ui settings", default='config.json')
		args, _ = parser.parse_known_args(sys.argv)

		sys.argv, _ = extract_arg(sys.argv, '-f')
		sys.argv, skip_torch_cuda_test = extract_arg(sys.argv, '--skip-torch-cuda-test')
		sys.argv, reinstall_xformers = extract_arg(sys.argv, '--reinstall-xformers')
		sys.argv, update_check = extract_arg(sys.argv, '--update-check')

webui.sh

+2 −1

Original line number	Diff line number	Diff line
		@@ -51,10 +51,11 @@ fi
		can_run_as_root=0

		# read any command line flags to the webui.sh script
		while getopts "f" flag
		while getopts "f" flag > /dev/null 2>&1
		do
		case ${flag} in
		f) can_run_as_root=1;;
		*) break;;
		esac
		done

Admin message