From 4affef59d2cb6edc0288090518ca8adf7667cbc3 Mon Sep 17 00:00:00 2001 From: yewentao Date: Fri, 14 Jun 2024 17:42:08 +0800 Subject: [PATCH] fix: do not need to register for windows(spawn) --- lazyllm/common/logger.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/lazyllm/common/logger.py b/lazyllm/common/logger.py index 7e4f0783..8a9cc15e 100644 --- a/lazyllm/common/logger.py +++ b/lazyllm/common/logger.py @@ -6,6 +6,7 @@ from sys import stderr from zipfile import ZipFile import lazyllm +import platform from .utils import check_path from loguru import logger @@ -116,7 +117,7 @@ def add_file_sink(): log_file_path = join(log_dir_path, log_file_name) LOG.add( log_file_path, - level=lazyllm.config['log_file_level'], + level=lazyllm.config["log_file_level"], format="{message}", encoding="utf-8", rotation=lazyllm.config["log_file_size"], @@ -132,6 +133,7 @@ def add_file_sink(): add_file_sink() -os.register_at_fork( - after_in_child=add_file_sink, -) +if platform.system() != "Windows": + os.register_at_fork( + after_in_child=add_file_sink, + )