diff --git a/CMD_FLAGS.txt b/CMD_FLAGS.txt index 47fe8405..e69de29b 100644 --- a/CMD_FLAGS.txt +++ b/CMD_FLAGS.txt @@ -1 +0,0 @@ ---listen diff --git a/start_linux.sh b/start_linux.sh index 1de8ccdf..6792902a 100755 --- a/start_linux.sh +++ b/start_linux.sh @@ -64,4 +64,4 @@ source "$CONDA_ROOT_PREFIX/etc/profile.d/conda.sh" # otherwise conda complains a conda activate "$INSTALL_ENV_DIR" # setup installer env -python webui.py +python webui.py $@ diff --git a/start_macos.sh b/start_macos.sh index 131a8af5..5f5bb439 100755 --- a/start_macos.sh +++ b/start_macos.sh @@ -64,4 +64,4 @@ source "$CONDA_ROOT_PREFIX/etc/profile.d/conda.sh" # otherwise conda complains a conda activate "$INSTALL_ENV_DIR" # setup installer env -python webui.py +python webui.py $@ diff --git a/start_windows.bat b/start_windows.bat index 3140f70f..77ba93bf 100755 --- a/start_windows.bat +++ b/start_windows.bat @@ -67,7 +67,7 @@ set "CUDA_HOME=%CUDA_PATH%" call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" || ( echo. && echo Miniconda hook not found. && goto end ) @rem setup installer env -call python webui.py +call python webui.py %* @rem below are functions for the script next line skips these during normal execution goto end diff --git a/start_wsl.bat b/start_wsl.bat index 41fa572f..26d2f8a6 100755 --- a/start_wsl.bat +++ b/start_wsl.bat @@ -5,7 +5,7 @@ cd /D "%~dp0" set PATH=%PATH%;%SystemRoot%\system32 @rem sed -i 's/\x0D$//' ./wsl.sh converts newlines to unix format in the wsl script -call wsl -e bash -lic "sed -i 's/\x0D$//' ./wsl.sh; source ./wsl.sh" +call wsl -e bash -lic "sed -i 's/\x0D$//' ./wsl.sh; source ./wsl.sh $@" :end pause diff --git a/webui.py b/webui.py index 4225cd7e..9d2fbaf1 100644 --- a/webui.py +++ b/webui.py @@ -245,7 +245,8 @@ def download_model(): def launch_webui(): - run_cmd(f"python server.py {CMD_FLAGS}", environment=True) + flags = [flag for flag in sys.argv[1:] if flag != '--update'] + run_cmd(f"python server.py {' '.join(flags)} {CMD_FLAGS}", environment=True) if __name__ == "__main__": @@ -254,7 +255,7 @@ if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument('--update', action='store_true', help='Update the web UI.') - args = parser.parse_args() + args, _ = parser.parse_known_args() if args.update: update_dependencies() diff --git a/wsl.sh b/wsl.sh index 6660bc2e..6b62adca 100755 --- a/wsl.sh +++ b/wsl.sh @@ -106,5 +106,5 @@ fi # setup installer env update env if called with 'wsl.sh update' case "$1" in ("update") python webui.py --update;; -(*) python webui.py;; +(*) python webui.py $@;; esac