From b912bdefa25e73c70f63ebe1370050efbdb829c4 Mon Sep 17 00:00:00 2001 From: digital-garbage <133010426+digital-garbage@users.noreply.github.com> Date: Sat, 30 May 2026 12:04:22 +0300 Subject: [PATCH] Add |loved variants to V2_RATING_LABELS for ComfyUI validation ComfyUI rejects widget values not in the allowed list. Generate all valid label|loved combinations at definition time. Co-Authored-By: Claude Sonnet 4.6 --- conditioning.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/conditioning.py b/conditioning.py index c2d49aa8..0eb68522 100644 --- a/conditioning.py +++ b/conditioning.py @@ -5630,7 +5630,7 @@ FunPackGemmaEmbeddingRefiner = FunPackVideoRefiner _V2_PERSISTENT_ENCODE_CACHE = {} _V2_PERSISTENT_CACHE_MAX = 4096 -V2_RATING_LABELS = [ +_BASE_RATING_LABELS = [ "-Just forget it-", "Perfect", "Nailed it", @@ -5648,6 +5648,10 @@ V2_RATING_LABELS = [ "Missing action + quality", "Awful", ] +_NO_LOVED_LABELS = {"-Just forget it-", "Perfect", "Awful"} +V2_RATING_LABELS = _BASE_RATING_LABELS + [ + l + "|loved" for l in _BASE_RATING_LABELS if l not in _NO_LOVED_LABELS +] V2_RATING_PROFILES = { "-Just forget it-": {"key": "forget", "reward": 0.0, "level": 0, "missing_axes": [], "skip_learning": True},