text-generation-webui/extensions/softprompt/script.py
2023-02-11 17:17:10 -03:00

28 lines
569 B
Python

params = {
"soft prompt": " *I speak in an annoyingly cute way*",
}
def input_modifier(string):
"""
This function is applied to your text inputs before
they are fed into the model.
"""
return string
def output_modifier(string):
"""
This function is applied to the model outputs.
"""
return string
def bot_prefix_modifier(string):
"""
This function is only applied in chat mode. It modifies
the prefix text for the Bot and can be used to bias its
behavior.
"""
return string + params["soft prompt"]