fix webui

Former-commit-id: 7152fde4a026e67f15885814c1900f3911d04ee8
2024-12-19 06:48:03 +00:00
parent 1898c1e9a6
commit d90b40b60f
2 changed files with 2 additions and 0 deletions
--- a/src/llamafactory/webui/chatter.py
+++ b/src/llamafactory/webui/chatter.py
@@ -91,6 +91,7 @@ class WebChatModel(ChatModel):
            rope_scaling=get("top.rope_scaling") if get("top.rope_scaling") in ["linear", "dynamic"] else None,
            infer_backend=get("infer.infer_backend"),
            infer_dtype=get("infer.infer_dtype"),
+            trust_remote_code=True,
        )

        if checkpoint_path: