From b2d12d04b4547e6ad7253e94931b156bbc68acc4 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Sun, 4 Aug 2024 15:53:52 +0900 Subject: [PATCH] fix: sample_with_custom_noise - move intermediate latent to intermediate device. --- modules/impact/config.py | 2 +- modules/impact/impact_sampling.py | 3 +-- pyproject.toml | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/modules/impact/config.py b/modules/impact/config.py index 0b147d4..7f7c365 100644 --- 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 index 3f7b666..151e6b1 100644 --- a/modules/impact/impact_sampling.py +++ b/modules/impact/impact_sampling.py @@ -141,8 +141,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) @@ -154,6 +152,7 @@ def sample_with_custom_noise(model, add_noise, noise_seed, cfg, positive, negati 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 index 491c6a1..1853182 100644 --- 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"]