From 678f933d382641933920e84414fe36f89d1da5a3 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sat, 6 May 2023 19:00:49 -0400
Subject: [PATCH] maximum_batch_area for xformers.

Remove useless code.
---
 comfy/model_management.py | 7 ++++++-
 nodes.py                  | 4 +---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 7070912..b0640d6 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -312,7 +312,12 @@ def maximum_batch_area():
         return 0
 
     memory_free = get_free_memory() / (1024 * 1024)
-    area = ((memory_free - 1024) * 0.9) / (0.6)
+    if xformers_enabled():
+        #TODO: this needs to be tweaked
+        area = 50 * memory_free
+    else:
+        #TODO: this formula is because AMD sucks and has memory management issues which might be fixed in the future
+        area = ((memory_free - 1024) * 0.9) / (0.6)
     return int(max(area, 0))
 
 def cpu_mode():
diff --git a/nodes.py b/nodes.py
index c2bc368..ca0769b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -105,15 +105,13 @@ class ConditioningSetArea:
 
     CATEGORY = "conditioning"
 
-    def append(self, conditioning, width, height, x, y, strength, min_sigma=0.0, max_sigma=99.0):
+    def append(self, conditioning, width, height, x, y, strength):
         c = []
         for t in conditioning:
             n = [t[0], t[1].copy()]
             n[1]['area'] = (height // 8, width // 8, y // 8, x // 8)
             n[1]['strength'] = strength
             n[1]['set_area_to_bounds'] = False
-            n[1]['min_sigma'] = min_sigma
-            n[1]['max_sigma'] = max_sigma
             c.append(n)
         return (c, )