Merge pull request #132 from pollockjj/RES4LYF

#119 and #130 - RES4LYF Sampling errors

Author: pollockjj

distorch_2.py

@@ -301,6 +301,8 @@ def new_partially_load(self, device_to, extra_memory=0, full_load=False, force_p
 
             self.model.current_weight_patches_uuid = self.patches_uuid
 
+            self.model.device = device_to
+
             logger.info("[MultiGPU DisTorch V2] DisTorch loading completed.")
             logger.info(f"[MultiGPU DisTorch V2] Total memory: {mem_counter / (1024 * 1024):.2f}MB")
 

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.5.9"
+version = "2.5.10"
 license = {file = "LICENSE"}
 
 [project.urls]