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

Handle multiline inputs

Browse Source
This commit is contained in:
oobabooga 2023-05-07 11:54:26 -03:00
parent a35a2fab02
commit 8216c8f732
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
extensions/superbig/script.py
View File

@ -125,10 +125,10 @@ def apply_settings(_chunk_count):
def input_modifier(string):
# Find the user input
pattern = re.compile(r"<\|begin-user-input\|>(.*?)<\|end-user-input\|>")
pattern = re.compile(r"<\|begin-user-input\|>(.*?)<\|end-user-input\|>", re.DOTALL)
match = re.search(pattern, string)
if match:
user_input = match.group(1)
user_input = match.group(1).strip()
else:
user_input = ''
@ -178,7 +178,9 @@ def ui():
<|injection-point|>
### Input:
<|begin-user-input|>What datasets are mentioned in the paper above?<|end-user-input|>
<|begin-user-input|>
What datasets are mentioned in the paper above?
<|end-user-input|>
### Response:
```