From cc9986d2686964f809fd2384f05c790c7084d8d2 Mon Sep 17 00:00:00 2001 From: tavi Date: Sun, 22 Feb 2026 00:13:57 +0000 Subject: [PATCH] fix: replace assert with if/raise for pre_filter_count validation assert is stripped under python -O; use explicit if/raise ValueError so the invariant check is always active. --- comfy/ldm/lightricks/model.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/comfy/ldm/lightricks/model.py b/comfy/ldm/lightricks/model.py index e1a541a3b..60d760d29 100644 --- a/comfy/ldm/lightricks/model.py +++ b/comfy/ldm/lightricks/model.py @@ -916,10 +916,11 @@ class LTXVModel(LTXBaseModel): guide_entries = kwargs.get("guide_attention_entries", None) if guide_entries: total_pfc = sum(e["pre_filter_count"] for e in guide_entries) - assert total_pfc == len(kf_grid_mask), ( - f"guide pre_filter_counts ({total_pfc}) != " - f"keyframe grid mask length ({len(kf_grid_mask)})" - ) + if total_pfc != len(kf_grid_mask): + raise ValueError( + f"guide pre_filter_counts ({total_pfc}) != " + f"keyframe grid mask length ({len(kf_grid_mask)})" + ) resolved_entries = [] offset = 0 for entry in guide_entries: