fix: sample_with_custom_noise - sd3 compatibility

ltdrdata · Jun 12, 2024 · 885ca3d · 885ca3d
1 parent 23b7e8f
commit 885ca3d
Show file tree

Hide file tree

Showing 3 changed files with 4 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 = [5, 12, 1]
+version_code = [5, 12, 2]
 version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 dependency_version = 21

diff --git a/modules/impact/impact_sampling.py b/modules/impact/impact_sampling.py
@@ -116,9 +116,9 @@ def sample_with_custom_noise(model, add_noise, noise_seed, cfg, positive, negati
 
     if noise is None:
         if not add_noise:
-            noise = Noise_EmptyNoise().generate_noise(latent)
+            noise = Noise_EmptyNoise().generate_noise({'samples': latent_image})
         else:
-            noise = Noise_RandomNoise(noise_seed).generate_noise(latent)
+            noise = Noise_RandomNoise(noise_seed).generate_noise({'samples': latent_image})
 
     noise_mask = None
     if "noise_mask" in latent:

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 = "5.12.1"
+version = "5.12.2"
 license = "LICENSE"
 dependencies = ["segment-anything", "scikit-image", "piexif", "transformers", "opencv-python-headless", "GitPython", "scipy>=1.11.4"]