comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-01 14:09:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Default ro smaller blocks getting flipflopped first

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-10-02 18:00:21 -07:00
parent d5001ed90e
commit 0d8e8abd90
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@@ -639,7 +639,7 @@ class ModelPatcher:
block_buffer = 3
valid_block_types = []
# for each block type, check if have enough room to flipflop
for block_info in self.model.diffusion_model.get_all_block_module_sizes(reverse_sort_by_size=False):
for block_info in self.model.diffusion_model.get_all_block_module_sizes(reverse_sort_by_size=True):
block_size: int = block_info[1]
if block_size * block_buffer < lowvram_model_memory:
valid_block_types.append(block_info)