diff --git a/pulling_ace/utils/subprocessor.py b/pulling_ace/utils/subprocessor.py index 6acbf01..eca1f82 100644 --- a/pulling_ace/utils/subprocessor.py +++ b/pulling_ace/utils/subprocessor.py @@ -240,14 +240,13 @@ def leakreplay(model_type, model_name, probe): def riskcard_wrapper(args): return riskcards(args["model_type"], args["model_name"], args["probe"]) - def promptinjection_wrapper(args): return promptinjection(args["model_type"], args["model_name"], args["probe"]) - def toxicity_wrapper(args): return toxicity(args["model_type"], args["model_name"], args["probe"]) + def leakreplay_wrapper(args): return toxicity(args["model_type"], args["model_name"], args["probe"])