From 1488f2c59bdfefbaf3c259a08ba434161013403b Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 31 Jan 2025 10:58:11 -0800
Subject: [PATCH] Logger should check attributes on current sys.stdout, which
 may have been overwritten

---
 comfy/app/logger.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/app/logger.py b/comfy/app/logger.py
index c5dc32308..78a8879f0 100644
--- a/comfy/app/logger.py
+++ b/comfy/app/logger.py
@@ -57,6 +57,10 @@ def setup_logger(log_level: str = 'INFO', capacity: int = 300, use_stdout: bool
     if logs:
         return
 
+    # workaround for google collab
+    if not hasattr(sys.stdout, "buffer") or not hasattr(sys.stdout, "encoding") or not hasattr(sys.stdout, "line_buffering"):
+        return
+
     # Override output streams and log to buffer
     logs = deque(maxlen=capacity)