add export_device in webui #3333

This commit is contained in:
hiyouga committed 2024-04-25 19:02:32 +08:00
1 parent a3aad4b4f0
commit 3a7c1286ce
7 files changed
+57 -22

No files matched your search

+1 -1
View File
@@ -139,7 +139,7 @@ class ModelArguments:
)
export_device: str = field(
default="cpu",
metadata={"help": "The device used in model export."},
metadata={"help": "The device used in model export, use cuda to avoid addmm errors."},
)
export_quantization_bit: Optional[int] = field(
default=None,