A logging filter would be more appropriate than the current async with approach for --comfyui-inference-log-level
|
async with temporary_log_level("comfy", self._comfyui_inference_log_level): |
|
out_tensor = await self.client.get_audio_output() |
|
parser.add_argument( |
|
"--comfyui-inference-log-level", |
|
default=None, |
|
choices=logging._nameToLevel.keys(), |
|
help="Set the logging level for ComfyUI inference", |
|
) |
|
args = parser.parse_args() |
|
|
|
# Allow overriding of ComyfUI log levels. |
|
if args.comfyui_log_level: |
|
log_level = logging._nameToLevel.get(args.comfyui_log_level.upper()) |
|
logging.getLogger("comfy").setLevel(log_level) |