fix: Taste steering is named "(REINS)" and says it moves toward liked AND away from disliked, only after 2+ of each

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
digital-garbage
2026-10-07 14:46:44 +03:00
co-authored by Claude Opus 5.5
parent 923664ad44
commit dc67fc8274
+5 -5
View File
@@ -18,7 +18,7 @@ import torch
from ..._core import dit_hooks, log, registry
ID = "reins"
TITLE = "Taste steering"
TITLE = "Taste steering (REINS)"
MOUNT = "generation.sampling"
STAGE = "guidance"
CATEGORY = "guidance"
@@ -29,8 +29,8 @@ USES = ["taste_store"]
SETTINGS = {
"enabled": {
"type": "bool", "default": False,
"label": "Taste steering",
"hint": "Learns from your ratings and nudges the picture toward what you liked.",
"label": "Taste steering (REINS)",
"hint": "REINS: at the chosen blocks, shifts the picture by (your liked runs − your disliked runs): toward liked AND away from disliked. Does nothing until the Taste key has 2+ liked and 2+ disliked runs. Picture only; sound follows.",
},
"strength": {
"type": "float", "default": 0.01, "min": 0.0, "max": 2.0, "step": 0.01,
@@ -50,7 +50,7 @@ KIND = "reins"
def _say(message):
log.once(f"{ID}:{message}", log.ALERT, "FunPack Taste steering", message)
log.once(f"{ID}:{message}", log.ALERT, "FunPack Taste steering (REINS)", message)
def install(patcher, values, key):
@@ -72,7 +72,7 @@ def install(patcher, values, key):
def fresh():
live["dirs"], summary = taste.directions(KIND, steer, strength)
cast.clear()
log.once(f"{ID}:state", log.INFO, "FunPack Taste steering", summary)
log.once(f"{ID}:state", log.INFO, "FunPack Taste steering (REINS)", summary)
captured = taste.collect(patcher, key, KIND, fresh=fresh)
spans = {}