Adjust memory usage factor calculation for flux2 klein. (#11900)

2026-01-16 09:10:50 +08:00 · 2026-01-15 17:06:40 -08:00 · 2026-01-15 17:06:40 -08:00 · 4c816d5c69
commit 4c816d5c69
parent 6125b3a5e7
1 changed files with 1 additions and 1 deletions
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@ -763,7 +763,7 @@ class Flux2(Flux):

    def __init__(self, unet_config):
        super().__init__(unet_config)
-        self.memory_usage_factor = self.memory_usage_factor * (2.0 * 2.0) * 2.36
+        self.memory_usage_factor = self.memory_usage_factor * (2.0 * 2.0) * (unet_config['hidden_size'] / 2604)

    def get_model(self, state_dict, prefix="", device=None):
        out = model_base.Flux2(self, device=device)