From 96d56d4f3c0b3b114032c10a50e090a8c158c045 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 10 Feb 2023 15:24:26 -0300 Subject: [PATCH] Turn the example script into a soft prompt script --- extensions/example/script.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/extensions/example/script.py b/extensions/example/script.py index fd303dcb..1892cbf6 100644 --- a/extensions/example/script.py +++ b/extensions/example/script.py @@ -1,6 +1,5 @@ params = { - "input suffix": " *I say as I make a funny face*", - "bot prefix": " *I speak in a cute way*", + "soft prompt": " *I speak in an annoyingly cute way*", } def input_modifier(string): @@ -9,7 +8,7 @@ def input_modifier(string): they are fed into the model. """ - return string + params["input suffix"] + return string def output_modifier(string): """ @@ -25,4 +24,4 @@ def bot_prefix_modifier(string): behavior. """ - return string + params["bot prefix"] + return string + params["soft prompt"]