423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-18 12:50:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

update readme, fix web ui postprocess

Browse Source
This commit is contained in:
hiyouga
2023-07-22 14:29:22 +08:00
parent 21daf34f0c
commit 035c966d5c
6 changed files with 81 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/llmtuner/webui/chat.py
View File

@@ -90,6 +90,8 @@ class WebChatModel(ChatModel):
yield chatbot, new_history
def postprocess(self, response: str) -> str:
response = response.replace("<", "&lt;")
response = response.replace(">", "&gt;")
return response
blocks = response.split("```")
for i, block in enumerate(blocks):
if i % 2 == 0:
blocks[i] = block.replace("<", "&lt;").replace(">", "&gt;")
return "```".join(blocks)