Tweak regular SD memory formula.

7 months ago · 3a9ee995cf
parent 47da42d928
commit 3a9ee995cf
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@ -252,7 +252,7 @@ class BaseModel(torch.nn.Module):
                dtype = self.manual_cast_dtype
            #TODO: this needs to be tweaked
            area = input_shape[0] * math.prod(input_shape[2:])
-            return (area * comfy.model_management.dtype_size(dtype) / 50) * (1024 * 1024)
+            return (area * comfy.model_management.dtype_size(dtype) * 0.01) * (1024 * 1024)
        else:
            #TODO: this formula might be too aggressive since I tweaked the sub-quad and split algorithms to use less memory.
            area = input_shape[0] * math.prod(input_shape[2:])