fix: impact_sampling - potential device mismatch

Co-authored-by: snomiao <snomiao+comfy-pr@gmail.com>

modules/impact/config.py

@@ -1,7 +1,7 @@
 import configparser
 import os
 
-version_code = [6, 0]
+version_code = [6, 0, 2]
 version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 dependency_version = 22

modules/impact/impact_sampling.py

@@ -6,6 +6,7 @@ import latent_preview
 import comfy
 import torch
 import math
+import comfy.model_management as mm
 
 
 try:
@@ -141,8 +142,6 @@ def sample_with_custom_noise(model, add_noise, noise_seed, cfg, positive, negati
         touched_callback = preview_callback
 
     disable_pbar = not comfy.utils.PROGRESS_BAR_ENABLED
-    # samples = comfy.sample.sample_custom(model, noise, cfg, sampler, sigmas, positive, negative, latent_image,
-    #                                      noise_mask=noise_mask, callback=touched_callback, disable_pbar=disable_pbar, seed=noise_seed)
 
     if negative != 'NegativePlaceholder':
         guider = comfy.samplers.CFGGuider(model)
@@ -153,7 +152,15 @@ def sample_with_custom_noise(model, add_noise, noise_seed, cfg, positive, negati
         positive = node_helpers.conditioning_set_values(positive, {"guidance": cfg})
         guider.set_conds(positive)
 
+    device = mm.get_torch_device()
+
+    noise = noise.to(device)
+    latent_image = latent_image.to(device)
+    if noise_mask is not None:
+        noise_mask = noise_mask.to(device)
+
     samples = guider.sample(noise, latent_image, sampler, sigmas, denoise_mask=noise_mask, callback=touched_callback, disable_pbar=disable_pbar, seed=noise_seed)
+    samples = samples.to(comfy.model_management.intermediate_device())
 
     out["samples"] = samples
     if "x0" in x0_output:

pyproject.toml

@@ -1,8 +1,8 @@
 [project]
 name = "comfyui-impact-pack"
 description = "This extension offers various detector nodes and detailer nodes that allow you to configure a workflow that automatically enhances facial details. And provide iterative upscaler."
-version = "6.0"
-license = "LICENSE"
+version = "6.0.2"
+license = { file = "LICENSE.txt" }
 dependencies = ["segment-anything", "scikit-image", "piexif", "transformers", "opencv-python-headless", "GitPython", "scipy>=1.11.4"]
 
 [project.urls]