Merge pull request #215 from Kosinkadink/develop

Fixed backwards compatibility for new KF scheduling behavior, changes…

Author: Kosinkadink

adv_control/utils.py

@@ -18,6 +18,7 @@
 
 BIGMIN = -(2**53-1)
 BIGMAX = (2**53-1)
+BIGMAX_TENSOR = torch.tensor(BIGMAX)
 
 ORIG_PREVIOUS_CONTROLNET = "_orig_previous_controlnet"
 CONTROL_INIT_BY_ACN = "_control_init_by_ACN"
@@ -562,7 +563,7 @@ def prepare_current_timestep(self, t: Tensor, transformer_options: dict[str, tor
         # get current step percent
         curr_t: float = self.t
         prev_index = self._current_timestep_index
-        max_sigma = torch.max(transformer_options.get("sigmas", BIGMAX))
+        max_sigma = torch.max(transformer_options.get("sample_sigmas", BIGMAX_TENSOR))
         # if met guaranteed steps (or no current keyframe), look for next keyframe in case need to switch
         if self._current_timestep_keyframe is None or self._current_used_steps >= self._current_timestep_keyframe.get_effective_guarantee_steps(max_sigma):
             # if has next index, loop through and see if need to switch

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-advanced-controlnet"
 description = "Nodes for scheduling ControlNet strength across timesteps and batched latents, as well as applying custom weights and attention masks."
-version = "1.5.0"
+version = "1.5.1"
 license = { file = "LICENSE" }
 dependencies = []