From 0ca771fe685b25fd07ed2b1d899384550553cc70 Mon Sep 17 00:00:00 2001
From: John Pollock <pollockjj@gmail.com>
Date: Thu, 28 Aug 2025 20:47:12 -0500
Subject: [PATCH] Fix for :
 https://github.com/pollockjj/ComfyUI-MultiGPU/issues/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
---
 distorch_2.py  | 2 ++
 pyproject.toml | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/distorch_2.py b/distorch_2.py
index 9725ad0..e66c7fa 100644
--- a/distorch_2.py
+++ b/distorch_2.py
@@ -624,6 +624,8 @@ def override_class_with_distorch_safetensor_v2(cls):
             vram_string = ""
             if virtual_vram_gb > 0:
                 vram_string = f"{compute_device};{virtual_vram_gb};{donor_device}"
+            elif expert_mode_allocations:  # Only include compute device if there's an expert string
+                vram_string = compute_device
 
             full_allocation = f"{expert_mode_allocations}#{vram_string}" if expert_mode_allocations or vram_string else ""
             
diff --git a/pyproject.toml b/pyproject.toml
index 5740591..3e1008d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-multigpu"
 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."
-version = "2.2.0"
+version = "2.2.1"
 license = {file = "LICENSE"}
 
 [project.urls]