Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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"]