fix: sample_with_custom_noise - move intermediate latent to intermedi…

…ate device.
ltdrdata · Aug 4, 2024 · b2d12d0 · b2d12d0
1 parent 26c7419
commit b2d12d0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/modules/impact/config.py b/modules/impact/config.py
@@ -1,7 +1,7 @@
 import configparser
 import os
 
-version_code = [6, 0]
+version_code = [6, 0, 1]
 version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 dependency_version = 22

diff --git a/modules/impact/impact_sampling.py b/modules/impact/impact_sampling.py
@@ -141,8 +141,6 @@ def touched_callback(step, x0, x, total_steps):
         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)
@@ -154,6 +152,7 @@ def touched_callback(step, x0, x, total_steps):
         guider.set_conds(positive)
 
     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:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [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"
+version = "6.0.1"
 license = { file = "LICENSE.txt" }
 dependencies = ["segment-anything", "scikit-image", "piexif", "transformers", "opencv-python-headless", "GitPython", "scipy>=1.11.4"]