mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Some minor formatting changes

Browse source
This commit is contained in:
oobabooga 2023-02-18 11:07:55 -03:00
parent 3e6a8ccdce
commit d58544a420
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
server.py
View file

@ -250,6 +250,8 @@ def fix_galactica(s):
s = s.replace(r'\(', r'$')
s = s.replace(r'\)', r'$')
s = s.replace(r'$$', r'$')
s = re.sub(r'\n', r'\n\n', s)
s = re.sub(r"\n{3,}", "\n\n", s)
return s
def get_max_prompt_length(tokens):
@ -755,7 +757,7 @@ def save_history(timestamp=True):
if not Path('logs').exists():
Path('logs').mkdir()
with open(Path(f'logs/{fname}'), 'w') as f:
f.write(json.dumps({'data': history['internal'], 'data_visible': history['visible']}))
f.write(json.dumps({'data': history['internal'], 'data_visible': history['visible']}, indent=2))
return Path(f'logs/{fname}')
def load_history(file, name1, name2):