Skip to content

Commit 0ca771f

Browse files
committed
Fix for : #93
Fix compute device inclusion in expert mode allocations Include compute device in vram_string when expert_mode_allocations is set but virtual_vram_gb is 0. This ensures the compute device is properly specified in the full allocation string for expert mode configurations without virtual VRAM. Bump version to 2.2.1
1 parent 2481b52 commit 0ca771f

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

distorch_2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -624,6 +624,8 @@ def override(self, *args, compute_device=None, virtual_vram_gb=4.0,
624624
vram_string = ""
625625
if virtual_vram_gb > 0:
626626
vram_string = f"{compute_device};{virtual_vram_gb};{donor_device}"
627+
elif expert_mode_allocations: # Only include compute device if there's an expert string
628+
vram_string = compute_device
627629

628630
full_allocation = f"{expert_mode_allocations}#{vram_string}" if expert_mode_allocations or vram_string else ""
629631

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[project]
22
name = "comfyui-multigpu"
33
description = "Provides a suite of custom nodes to manage multiple GPUs for ComfyUI, including advanced model offloading for both GGUF and Safetensor formats with DisTorch, and bespoke MultiGPU support for WanVideoWrapper and other custom nodes."
4-
version = "2.2.0"
4+
version = "2.2.1"
55
license = {file = "LICENSE"}
66

77
[project.urls]

0 commit comments

Comments
 (0)