From da2629c12c145140190d52aa39a4f0f8ef48ee89 Mon Sep 17 00:00:00 2001
From: "codeflash-ai[bot]"
 <148906541+codeflash-ai[bot]@users.noreply.github.com>
Date: Tue, 22 Apr 2025 00:55:23 +0000
Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20Speed=20up=20function=20`a?=
 =?UTF-8?q?dd=5Farea=5Fdims`=20by=2056%=20To=20optimize=20the=20given=20fu?=
 =?UTF-8?q?nction,=20we=20can=20avoid=20repeated=20slicing=20and=20concate?=
 =?UTF-8?q?nations=20within=20the=20loop,=20which=20can=20be=20computation?=
 =?UTF-8?q?ally=20expensive,=20especially=20for=20large=20lists.=20Instead?=
 =?UTF-8?q?,=20we=20can=20split=20the=20list=20just=20once=20and=20constru?=
 =?UTF-8?q?ct=20the=20final=20result=20using=20list=20operations=20more=20?=
 =?UTF-8?q?efficiently.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Here's the optimized version of the program.



### Optimizations.
1. Calculate the current number of dimensions (`current_dims`) once before the loop.
2. Within the loop, use the `extend()` method to append parts of the `area` list efficiently rather than using concatenation (`+`) multiple times.
3. Use a single list construction operation to build the new area list in-place.

This avoids the repeated creation of intermediary lists and makes the loop more efficient.
---
 comfy/samplers.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/comfy/samplers.py b/comfy/samplers.py
index 27dfce45a..178a9b87e 100644
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -21,8 +21,16 @@ import numpy
 
 
 def add_area_dims(area, num_dims):
-    while (len(area) // 2) < num_dims:
-        area = [2147483648] + area[:len(area) // 2] + [0] + area[len(area) // 2:]
+    current_dims = len(area) // 2
+    while current_dims < num_dims:
+        midpoint = len(area) // 2
+        # More efficient construction of the new area list
+        new_area = [2147483648]
+        new_area.extend(area[:midpoint])
+        new_area.append(0)
+        new_area.extend(area[midpoint:])
+        area = new_area
+        current_dims += 1
     return area
 
 def get_area_and_mult(conds, x_in, timestep_in):