fix: KSampler Progress - 1st latent shape mismatch if flux/sd3

#161
ltdrdata · Sep 15, 2024 · c0d54d8 · c0d54d8
1 parent 74b2856
commit c0d54d8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/__init__.py b/__init__.py
@@ -7,7 +7,7 @@
 
 import importlib
 
-version_code = [1, 2]
+version_code = [1, 2, 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/sampler_nodes.py b/inspire/sampler_nodes.py
@@ -44,7 +44,7 @@ def doit(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, l
         if omit_start_latent:
             result = []
         else:
-            result = [latent_image['samples']]
+            result = [comfy.sample.fix_empty_latent_channels(model, latent_image['samples']).cpu()]
 
         def progress_callback(step, x0, x, total_steps):
             if (total_steps-1) != step and step % interval != 0:

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 = "1.2"
+version = "1.2.1"
 license = { file = "LICENSE" }
 dependencies = ["matplotlib", "cachetools"]