add export_device in webui #3333
This commit is contained in:
1 parent
a3aad4b4f0
commit
3a7c1286ce
7 files changed
+57
-22
No files matched your search
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user