update 1.2.0

pull/165/head
Akiba 2023-06-01 20:45:14 +08:00
parent 2eca4237ba
commit 732bc85058
No known key found for this signature in database
GPG Key ID: 9D600258808ACBCD
2 changed files with 3 additions and 2 deletions

@ -1 +1 @@
Subproject commit 5934ccccb82da56de38b19b1089b65f3448e773c
Subproject commit c6591cf348c4a276bd359500716cab815bcd71b4

3
gui.py
View File

@ -13,6 +13,7 @@ from typing import List
parser = argparse.ArgumentParser(description="GUI for stable diffusion training")
parser.add_argument("--host", type=str, default="127.0.0.1")
parser.add_argument("--port", type=int, default=28000, help="Port to run the server on")
parser.add_argument("--tensorboard-host", type=str, default="127.0.0.1", help="Port to run the tensorboard")
parser.add_argument("--tensorboard-port", type=int, default=6006, help="Port to run the tensorboard")
parser.add_argument("--dev", action="store_true")
@ -54,7 +55,7 @@ def remove_warnings():
def run_tensorboard():
print("Starting tensorboard...")
subprocess.Popen([sys.executable, "-m", "tensorboard.main", "--logdir", "logs", "--port", str(args.tensorboard_port)])
subprocess.Popen([sys.executable, "-m", "tensorboard.main", "--logdir", "logs", "--host", args.tensorboard_host, "--port", str(args.tensorboard_port)])
def check_dirs(dirs: List):