wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-06-11 16:57:29 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Reduce memory consumption in model_detection tests.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-02-26 22:13:24 +01:00
parent 0ee3231db2
commit d7eb2ac8cc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests-unit/comfy_test/model_detection_test.py
View File

@ -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)