From 47da42d9283815a58636bd6b42c0434f70b24c9c Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 2 Aug 2024 17:02:35 -0400 Subject: [PATCH] Better Flux vram estimation. --- comfy/model_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_base.py b/comfy/model_base.py index 994b414..9989a4c 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -710,7 +710,7 @@ class Flux(BaseModel): 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)