turboderp-org/exllamav2
1
0
Fork 0
mirror of https://github.com/turboderp-org/exllamav2.git synced 2026-04-20 06:19:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed padding issue

Browse Source
This commit is contained in:
SinanAkkoyun
2023-11-29 01:43:20 +01:00
parent a58c162de7
commit cc1ca45250
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/chat_formatting.py
View File

@@ -63,8 +63,8 @@ class BlackBackgroundTerminalFormatter(TerminalFormatter):
# Padding of code
lines = content.split('\n')
padded_lines = [f"{lines[0]}{' ' * self.code_pad * 2}"] + [f"{' ' * self.code_pad}{line}{' ' * self.code_pad}" for line in
lines[1:-1]] + [lines[-1]]
padded_lines = [f"{' ' * self.code_pad}{line}{' ' * self.code_pad}" for line in lines][:-1]
content = '\n'.join(padded_lines)
# Modify the ANSI codes to include a black background
@@ -73,6 +73,9 @@ class BlackBackgroundTerminalFormatter(TerminalFormatter):
# Offset codeblock
modified_content = '\n'.join([modified_content.split('\n')[0]] + [f"{' ' * self.block_pad_left}{line}" for line in
modified_content.split('\n')[1:]])
# add newline
modified_content += '\n'
# Relay the modified content to the outfile
outfile.write(modified_content)