From d7eb2ac8cc5d9de3aa62f80768393cf9a8d4c78e Mon Sep 17 00:00:00 2001 From: Talmaj Marinc Date: Thu, 26 Feb 2026 22:13:24 +0100 Subject: [PATCH] Reduce memory consumption in model_detection tests. --- tests-unit/comfy_test/model_detection_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests-unit/comfy_test/model_detection_test.py b/tests-unit/comfy_test/model_detection_test.py index fdde8994a..2551a417b 100644 --- a/tests-unit/comfy_test/model_detection_test.py +++ b/tests-unit/comfy_test/model_detection_test.py @@ -7,7 +7,7 @@ import comfy.supported_models def _make_longcat_comfyui_sd(): """Minimal ComfyUI-format state dict for pre-converted LongCat-Image weights.""" sd = {} - H = 3072 + H = 32 # Reduce hidden state dimension to reduce memory usage C_IN = 16 C_CTX = 3584 @@ -39,7 +39,7 @@ def _make_longcat_comfyui_sd(): def _make_flux_schnell_comfyui_sd(): """Minimal ComfyUI-format state dict for standard Flux Schnell.""" sd = {} - H = 3072 + H = 32 # Reduce hidden state dimension to reduce memory usage C_IN = 16 sd["img_in.weight"] = torch.empty(H, C_IN * 4)