feat: re-implement abliteration as a modifier plugin

This commit is contained in:
Philipp Emanuel Weidmann
2026-09-21 18:44:54 +05:30
parent 92ab7f09d5
commit 598f5f4bd4
24 changed files with 1004 additions and 1726 deletions
+13 -13
View File
@@ -1,4 +1,4 @@
# This test case is for Dense models.
# This test case is for dense models.
# After any change related to it, this test should PASS.
model = "tiny-random/mistral-3"
@@ -18,18 +18,6 @@ trial_index = 0
model_action = "save"
save_directory = "model"
[good_prompts]
dataset = "mlabonne/harmless_alpaca"
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
split = "train[:5]"
column = "text"
[bad_prompts]
dataset = "mlabonne/harmful_behaviors"
commit = "01cead01398926d81f7c52bdb790ee8cf77ebba7"
split = "train[:5]"
column = "text"
[scorer.KLDivergence.prompts]
dataset = "mlabonne/harmless_alpaca"
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
@@ -41,3 +29,15 @@ dataset = "mlabonne/harmful_behaviors"
commit = "01cead01398926d81f7c52bdb790ee8cf77ebba7"
split = "test[:5]"
column = "text"
[modifier.Abliteration.good_prompts]
dataset = "mlabonne/harmless_alpaca"
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
split = "train[:5]"
column = "text"
[modifier.Abliteration.bad_prompts]
dataset = "mlabonne/harmful_behaviors"
commit = "01cead01398926d81f7c52bdb790ee8cf77ebba7"
split = "train[:5]"
column = "text"