Skip to content

Commit 3a9ee99

Browse files
Tweak regular SD memory formula.
1 parent 47da42d commit 3a9ee99

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,7 @@ def memory_required(self, input_shape):
252252
dtype = self.manual_cast_dtype
253253
#TODO: this needs to be tweaked
254254
area = input_shape[0] * math.prod(input_shape[2:])
255-
return (area * comfy.model_management.dtype_size(dtype) / 50) * (1024 * 1024)
255+
return (area * comfy.model_management.dtype_size(dtype) * 0.01) * (1024 * 1024)
256256
else:
257257
#TODO: this formula might be too aggressive since I tweaked the sub-quad and split algorithms to use less memory.
258258
area = input_shape[0] * math.prod(input_shape[2:])

0 commit comments

Comments
 (0)