From d150ca3adf2427202e2920c350aae2c1888a628b Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Sat, 22 Jun 2024 21:25:40 +0900
Subject: [PATCH] hotfix: Scheduled CFGGuider - invalid fallback handling

---
 __init__.py              | 2 +-
 inspire/sampler_nodes.py | 7 ++++++-
 pyproject.toml           | 2 +-
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 7765070..e167f52 100644
--- a/__init__.py
+++ b/__init__.py
@@ -7,7 +7,7 @@
 
 import importlib
 
-version_code = [0, 80, 2]
+version_code = [0, 80, 3]
 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
index 25727ef..cb952b4 100644
--- a/inspire/sampler_nodes.py
+++ b/inspire/sampler_nodes.py
@@ -175,6 +175,7 @@ def __init__(self, model_patcher, sigmas, from_cfg, to_cfg, schedule):
         self.from_cfg = from_cfg
         self.to_cfg = to_cfg
         self.schedule = schedule
+        self.last_i = 0
         self.renew_cfg_sigmas()
 
     def set_cfg(self, cfg):
@@ -199,6 +200,8 @@ def renew_cfg_sigmas(self):
             self.cfg_sigmas_i[i] = self.cfg_sigmas[k]
             i += 1
 
+        print(f"self.cfg_sigmas: {self.cfg_sigmas}")
+
     def predict_noise(self, x, timestep, model_options={}, seed=None):
         k = float(timestep[0])
 
@@ -206,6 +209,7 @@ def predict_noise(self, x, timestep, model_options={}, seed=None):
         if v is None:
             # fallback
             v = self.cfg_sigmas_i[self.last_i+1]
+            self.cfg_sigmas[k] = v
 
         self.last_i = v[1]
         self.cfg = v[0]
@@ -224,8 +228,8 @@ def __init__(self, model_patcher, sigmas, from_cfg, to_cfg, schedule, neg_scale)
         self.to_cfg = to_cfg
         self.schedule = schedule
         self.neg_scale = neg_scale
-        self.renew_cfg_sigmas()
         self.last_i = 0
+        self.renew_cfg_sigmas()
 
     def set_cfg(self, cfg):
         self.default_cfg = cfg
@@ -256,6 +260,7 @@ def predict_noise(self, x, timestep, model_options={}, seed=None):
         if v is None:
             # fallback
             v = self.cfg_sigmas_i[self.last_i+1]
+            self.cfg_sigmas[k] = v
 
         self.last_i = v[1]
         self.cfg = v[0]
diff --git a/pyproject.toml b/pyproject.toml
index e052951..cf9d2b8 100644
--- 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.80.2"
+version = "0.80.3"
 license = "LICENSE"
 dependencies = ["matplotlib", "cachetools"]