mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Turn the example script into a soft prompt script

Browse source
This commit is contained in:
oobabooga 2023-02-10 15:24:26 -03:00
parent e0b164feab
commit 96d56d4f3c
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
extensions/example/script.py
View file

@ -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"]