Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Change the injection context string

Browse Source
This commit is contained in:
oobabooga 2023-05-13 14:23:02 -03:00
parent 7cc17e3f1f
commit 5f6cf39f36
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/superbooga/script.py
View File

@ -90,7 +90,7 @@ def custom_generate_chat_prompt(user_input, state, **kwargs):
if state['mode'] == 'instruct': if state['mode'] == 'instruct':
results = collector.get(user_input, n_results=chunk_count) results = collector.get(user_input, n_results=chunk_count)
additional_context = '\nConsider the excerpts below as additional context:\n\n' + '\n'.join(results) additional_context = '\nYour reply should be based on the context below:\n\n' + '\n'.join(results)
user_input += additional_context user_input += additional_context
else: else: