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},