fix #4731

Former-commit-id: 99e016ee552a551b52b6fcf3616cb57a5b927715
2024-07-10 11:32:36 +08:00
parent 446129ca7a
commit 025c8fe302
2 changed files with 11 additions and 6 deletions
--- a/src/llamafactory/webui/components/export.py
+++ b/src/llamafactory/webui/components/export.py
@@ -48,7 +48,7 @@ def save_model(
    template: str,
    visual_inputs: bool,
    export_size: int,
-    export_quantization_bit: int,
+    export_quantization_bit: str,
    export_quantization_dataset: str,
    export_device: str,
    export_legacy_format: bool,