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 = { params = {
"input suffix": " *I say as I make a funny face*", "soft prompt": " *I speak in an annoyingly cute way*",
"bot prefix": " *I speak in a cute way*",
} }
def input_modifier(string): def input_modifier(string):
@ -9,7 +8,7 @@ def input_modifier(string):
they are fed into the model. they are fed into the model.
""" """
return string + params["input suffix"] return string
def output_modifier(string): def output_modifier(string):
""" """
@ -25,4 +24,4 @@ def bot_prefix_modifier(string):
behavior. behavior.
""" """
return string + params["bot prefix"] return string + params["soft prompt"]