Better Flux vram estimation.

NeoWorldTeam · Aug 2, 2024 · 47da42d · 47da42d
1 parent 17bbd83
commit 47da42d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/model_base.py b/comfy/model_base.py
@@ -710,7 +710,7 @@ def memory_required(self, input_shape):
                 dtype = self.manual_cast_dtype
             #TODO: this probably needs to be tweaked
             area = input_shape[0] * input_shape[2] * input_shape[3]
-            return (area * comfy.model_management.dtype_size(dtype) * 0.020) * (1024 * 1024)
+            return (area * comfy.model_management.dtype_size(dtype) * 0.026) * (1024 * 1024)
         else:
             area = input_shape[0] * input_shape[2] * input_shape[3]
             return (area * 0.3) * (1024 * 1024)