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

update readme, fix web ui postprocess

Browse Source 
Former-commit-id: 035c966d5c
This commit is contained in:
hiyouga
2023-07-22 14:29:22 +08:00
parent c76caa860f
commit 261ca840d0
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)

2
src/llmtuner/webui/interface.py
View File

@@ -55,4 +55,4 @@ def create_ui() -> gr.Blocks:
if __name__ == "__main__":
demo = create_ui()
demo.queue()
demo.launch(server_name="0.0.0.0", share=False, inbrowser=True)
demo.launch(server_name="0.0.0.0", server_port=7860, share=False, inbrowser=True)