From cd3a7bdb5ee3a37265e7c426bd9fbc88d95c5863 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Fri, 20 Sep 2024 20:34:49 +0000 Subject: [PATCH] Assert that change_ratio is in the expected range in TrajectoryGuidanceExtension. --- invokeai/backend/flux/trajectory_guidance_extension.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/invokeai/backend/flux/trajectory_guidance_extension.py b/invokeai/backend/flux/trajectory_guidance_extension.py index b6329a1b6b..333dea2194 100644 --- a/invokeai/backend/flux/trajectory_guidance_extension.py +++ b/invokeai/backend/flux/trajectory_guidance_extension.py @@ -91,6 +91,9 @@ class TrajectoryGuidanceExtension: x1=1.0, y1=self._change_ratio_at_t_1, x2=self._t_cutoff, y2=self._change_ratio_at_cutoff )(t_prev) + # The change_ratio should be in the range [0, 1]. Assert that we didn't make any mistakes. + eps = 1e-5 + assert 0.0 - eps <= change_ratio <= 1.0 + eps return change_ratio def step(