fix critical safenudge bug

soham-chitnis10 · soham-chitnis10 · commit 3879b21623c3 · 2026-05-07T01:37:11.000-04:00
diff --git a/api/safenudge.py b/api/safenudge.py
@@ -238,7 +238,7 @@ def generate_moderated(
                         >= tau
                     )
                 ):
-                    nudge_ids = self.tokenizer(SafeNudge.NUDGE + sentence)["input_ids"][1:]
+                    nudge_ids = self.tokenizer(SafeNudge.NUDGE)["input_ids"][1:]
                     nudge_piece = torch.tensor(
                         nudge_ids,
                         device=device,

Original file line number	Diff line number	Diff line change
`@@ -238,7 +238,7 @@ def generate_moderated(`
`238`	`238`	`>= tau`
`239`	`239`	`)`
`240`	`240`	`):`
`241`		`- nudge_ids = self.tokenizer(SafeNudge.NUDGE + sentence)["input_ids"][1:]`
	`241`	`+ nudge_ids = self.tokenizer(SafeNudge.NUDGE)["input_ids"][1:]`
`242`	`242`	`nudge_piece = torch.tensor(`
`243`	`243`	`nudge_ids,`
`244`	`244`	`device=device,`