Skip to content

Commit

Permalink
add --dump-sysinfo, a cmd arg to dump limited sysinfo file at startup
Browse files Browse the repository at this point in the history
  • Loading branch information
AUTOMATIC1111 committed Aug 30, 2023
1 parent 8d54739 commit d0026da
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 0 deletions.
7 changes: 7 additions & 0 deletions launch.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,13 @@


def main():
if args.dump_sysinfo:
filename = launch_utils.dump_sysinfo()

print(f"Sysinfo saved as {filename}. Exiting...")

exit(0)

launch_utils.startup_timer.record("initial startup")

with launch_utils.startup_timer.subcategory("prepare environment"):
Expand Down
1 change: 1 addition & 0 deletions modules/cmd_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
parser.add_argument("--log-startup", action='store_true', help="launch.py argument: print a detailed log of what's happening at startup")
parser.add_argument("--skip-prepare-environment", action='store_true', help="launch.py argument: skip all environment preparation")
parser.add_argument("--skip-install", action='store_true', help="launch.py argument: skip installation of packages")
parser.add_argument("--dump-sysinfo", action='store_true', help="launch.py argument: dump limited sysinfo file (without information about extensions, options) to disk and quit")
parser.add_argument("--loglevel", type=str, help="log level; one of: CRITICAL, ERROR, WARNING, INFO, DEBUG", default=None)
parser.add_argument("--do-not-download-clip", action='store_true', help="do not download CLIP model even if it's not included in the checkpoint")
parser.add_argument("--data-dir", type=str, default=os.path.dirname(os.path.dirname(os.path.realpath(__file__))), help="base path where all user data is stored")
Expand Down
13 changes: 13 additions & 0 deletions modules/launch_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,3 +434,16 @@ def start():
webui.api_only()
else:
webui.webui()


def dump_sysinfo():
from modules import sysinfo
import datetime

text = sysinfo.get()
filename = f"sysinfo-{datetime.datetime.utcnow().strftime('%Y-%m-%d-%H-%M')}.txt"

with open(filename, "w", encoding="utf8") as file:
file.write(text)

return filename

0 comments on commit d0026da

Please sign in to comment.