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 <noreply@anthropic.com>
This commit is contained in:
digital-garbage
2026-05-30 12:04:22 +03:00
co-authored by Claude Sonnet 4.6
parent bbfe1201e2
commit b912bdefa2
+5 -1
View File
@@ -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},