Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

web: allow custom host/port, default to 127.0.0.1 for security reasons

Browse Source
This commit is contained in:
Cragin Godley 2022-09-02 10:43:32 -04:00
parent ad6cf6f2f7
commit 68f62c8352
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/dream.py
View File

@ -99,7 +99,7 @@ def main():
cmd_parser = create_cmd_parser() cmd_parser = create_cmd_parser()
if opt.web: if opt.web:
dream_server_loop(t2i) dream_server_loop(t2i, opt.host, opt.port)
else: else:
main_loop(t2i, opt.outdir, opt.prompt_as_dir, cmd_parser, infile) main_loop(t2i, opt.outdir, opt.prompt_as_dir, cmd_parser, infile)
@ -270,7 +270,7 @@ def get_next_command(infile=None) -> str: #command string
print(f'#{command}') print(f'#{command}')
return command return command
def dream_server_loop(t2i): def dream_server_loop(t2i, host, port):
print('\n* --web was specified, starting web server...') print('\n* --web was specified, starting web server...')
# Change working directory to the stable-diffusion directory # Change working directory to the stable-diffusion directory
os.chdir( os.chdir(
@ -279,9 +279,9 @@ def dream_server_loop(t2i):
# Start server # Start server
DreamServer.model = t2i DreamServer.model = t2i
dream_server = ThreadingDreamServer(("0.0.0.0", 9090)) dream_server = ThreadingDreamServer((host, port))
print("\nStarted Stable Diffusion dream server!") print("\nStarted Stable Diffusion dream server!")
print("Point your browser at http://localhost:9090 or use the host's DNS name or IP address.") print(f"Point your browser at http://{host}:{port} or use the host's DNS name or IP address.")
try: try:
dream_server.serve_forever() dream_server.serve_forever()
@ -417,6 +417,18 @@ def create_argv_parser():
action='store_true', action='store_true',
help='Start in web server mode.', help='Start in web server mode.',
) )
parser.add_argument(
'--host',
type=str,
default='127.0.0.1',
help='Web server: Host or IP to listen on'
)
parser.add_argument(
'--port',
type=int,
default='9090',
help='Web server: Port to listen on'
)
parser.add_argument( parser.add_argument(
'--weights', '--weights',
default='model', default='model',