Add prompt hijack option for extensions

This commit is contained in:
oobabooga 2023-02-25 00:49:18 -03:00
parent fd19e4812f
commit 111b5d42e7
2 changed files with 27 additions and 8 deletions

View file

@ -16,6 +16,11 @@ input_hijack = {
'value': ["", ""]
}
prompt_hijack = {
'state': False,
'value': ""
}
def generate_chat_picture(picture, name1, name2):
text = f'*{name1} sends {name2} a picture that contains the following: "{caption_image(picture)}"*'
buffer = BytesIO()