FIXED: mix_noise - device mismatch error

ltdrdata · Aug 20, 2024 · b09b295 · b09b295
1 parent a15d336
commit b09b295
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/__init__.py b/__init__.py
@@ -7,7 +7,7 @@
 
 import importlib
 
-version_code = [0, 86]
+version_code = [0, 86, 1]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 print(f"### Loading: ComfyUI-Inspire-Pack ({version_str})")
 

diff --git a/inspire/libs/utils.py b/inspire/libs/utils.py
@@ -56,6 +56,8 @@ def slerp(val, low, high):
 
 
 def mix_noise(from_noise, to_noise, strength, variation_method):
+    to_noise = to_noise.to(from_noise.device)
+
     if variation_method == 'slerp':
         mixed_noise = slerp(strength, from_noise, to_noise)
     else:

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-inspire-pack"
 description = "This extension provides various nodes to support Lora Block Weight and the Impact Pack. Provides many easily applicable regional features and applications for Variation Seed."
-version = "0.86"
+version = "0.86.1"
 license = { file = "LICENSE" }
 dependencies = ["matplotlib", "cachetools"]