text-generation-webui/extensions/example/script.py
oobabooga 6da94e358c Add support for extensions parameters
Still experimental
2023-01-28 23:00:51 -03:00

19 lines
363 B
Python

params = {
"input suffix": " *I say as I make a funny face*",
}
def input_modifier(string):
"""
This function is applied to your text inputs before
they are fed into the model.
"""
return string + params["input suffix"]
def output_modifier(string):
"""
This function is applied to the model outputs.
"""
return string