fix webui

Former-commit-id: 7152fde4a026e67f15885814c1900f3911d04ee8
2025-10-14 15:52:49 +08:00 · 2024-12-19 06:48:03 +00:00 · 2024-12-19 06:48:03 +00:00 · d90b40b60f
commit d90b40b60f
parent 1898c1e9a6
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:
--- a/src/llamafactory/webui/components/export.py
+++ b/src/llamafactory/webui/components/export.py
@ -84,6 +84,7 @@ def save_model(
        export_quantization_dataset=export_quantization_dataset,
        export_device=export_device,
        export_legacy_format=export_legacy_format,
+        trust_remote_code=True,
    )

    if checkpoint_path: