mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Allow start scripts to have command-line flags

Browse source
This commit is contained in:
oobabooga 2023-09-22 10:03:56 -07:00
parent 8ab3eca9ec
commit 3314b7d795
7 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

5
webui.py
View file

@ -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()