From d9786583015d2a584663238ba1a2679453eee8eb Mon Sep 17 00:00:00 2001 From: BennyKok Date: Thu, 11 Jan 2024 12:38:54 +0800 Subject: [PATCH] fix(builder): logging issues --- .../modal-builder/src/template/data/install_deps.py | 2 ++ .../src/template/data/restore_snapshot.py | 11 ++++++----- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/builder/modal-builder/src/template/data/install_deps.py b/builder/modal-builder/src/template/data/install_deps.py index b90a8a5..3ff3ca3 100644 --- a/builder/modal-builder/src/template/data/install_deps.py +++ b/builder/modal-builder/src/template/data/install_deps.py @@ -3,6 +3,8 @@ import requests import time import subprocess +print("Starting ComfyUI") + command = ["python", "main.py", "--disable-auto-launch", "--disable-metadata", "--cpu"] # Start the server server_process = subprocess.Popen(command, cwd="/comfyui") diff --git a/builder/modal-builder/src/template/data/restore_snapshot.py b/builder/modal-builder/src/template/data/restore_snapshot.py index 27d898c..4ddbebe 100644 --- a/builder/modal-builder/src/template/data/restore_snapshot.py +++ b/builder/modal-builder/src/template/data/restore_snapshot.py @@ -6,7 +6,8 @@ import asyncio import logging # Set up the logger -logging.basicConfig(level=logging.INFO) +logging.basicConfig(level=logging.INFO, format='%(message)s') +logger = logging.getLogger("restore_snapshot") # Start the server # server_process = subprocess.Popen(command, cwd="/comfyui", stdout=subprocess.PIPE) @@ -47,19 +48,19 @@ async def read_stream(stream, isStderr): continue if not isStderr: - logging.info(l) + logger.info(l) # If the output matches one of the messages, print it and break the loop if success_message in l: - logging.info("Snapshot restore succeeded.") + logger.info("Snapshot restore succeeded.") break elif failure_message in l: - logging.info("Snapshot restore failed.") + logger.info("Snapshot restore failed.") break else: # is error # logger.error(l) - logging.error(l) + logger.error(l) break else: