upstream-archive/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-10-03 19:38:19 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix crash when selecting int8 or int4 cache in qwen image 2.1 (#16667)

Browse Source
This commit is contained in:
comfyanonymous
2026-09-30 00:00:09 -04:00
committed by GitHub
parent e6beab52ef
commit fb2315f11d
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/qwen_image21/model.py
+3
View File
@@ -395,6 +395,9 @@ class QwenImage21Transformer2DModel(nn.Module):
hidden_states = block(hidden_states, mod, pe, attn_fn, prefix_len, transformer_options)
for p in patches.get("single_block", []):
hidden_states = p({"img": hidden_states, "x": x, "block_index": i, "transformer_options": transformer_options})["img"]
if cache is not None:
# Release dequantized K/V before leaving the block's allocation scope.
del attn_fn, prefix_k, prefix_v
comfy.model_prefetch.prefetch_queue_pop(prefetch_queue, x.device, None, malloc_scope="block")
comfy.model_prefetch.malloc_graph_end()