From 31ca603ccbc45ab4db1279aa485c715b96b2aae8 Mon Sep 17 00:00:00 2001
From: Lucas - BLOCK33 <95554128+tonynoce@users.noreply.github.com>
Date: Sat, 21 Jun 2025 00:04:55 -0300
Subject: [PATCH 1/2] Improve the log time function for 10 minute + renders
 (#6207)

* modified:   main.py

* Update main.py
---
 main.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index c8c4194d4..79a652578 100644
--- a/main.py
+++ b/main.py
@@ -185,7 +185,13 @@ def prompt_worker(q, server_instance):
 
             current_time = time.perf_counter()
             execution_time = current_time - execution_start_time
-            logging.info("Prompt executed in {:.2f} seconds".format(execution_time))
+            
+            # Log Time in a more readable way after 10 minutes
+            if execution_time > 600:
+                execution_time = time.strftime("%H:%M:%S", time.gmtime(execution_time))
+                logging.info(f"Prompt executed in {execution_time}")
+            else:
+                logging.info("Prompt executed in {:.2f} seconds".format(execution_time))
 
         flags = q.get_flags()
         free_memory = flags.get("free_memory", False)

From 1883e70b4374a3317e0463a0bef292fc21182bad Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Sat, 21 Jun 2025 00:30:39 -0700
Subject: [PATCH 2/2] Fix exception when using a noise mask with cosmos
 predict2. (#8621)

* Fix exception when using a noise mask with cosmos predict2.

* Fix ruff.
---
 comfy/model_base.py | 2 ++
 main.py             | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index cb7689e84..75ec42699 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1024,6 +1024,8 @@ class CosmosPredict2(BaseModel):
     def process_timestep(self, timestep, x, denoise_mask=None, **kwargs):
         if denoise_mask is None:
             return timestep
+        if denoise_mask.ndim <= 4:
+            return timestep
         condition_video_mask_B_1_T_1_1 = denoise_mask.mean(dim=[1, 3, 4], keepdim=True)
         c_noise_B_1_T_1_1 = 0.0 * (1.0 - condition_video_mask_B_1_T_1_1) + timestep.reshape(timestep.shape[0], 1, 1, 1, 1) * condition_video_mask_B_1_T_1_1
         out = c_noise_B_1_T_1_1.squeeze(dim=[1, 3, 4])
diff --git a/main.py b/main.py
index 79a652578..0d7c97dcb 100644
--- a/main.py
+++ b/main.py
@@ -185,7 +185,7 @@ def prompt_worker(q, server_instance):
 
             current_time = time.perf_counter()
             execution_time = current_time - execution_start_time
-            
+
             # Log Time in a more readable way after 10 minutes
             if execution_time > 600:
                 execution_time = time.strftime("%H:%M:%S", time.gmtime(execution_time))