Merge branch 'hiyouga:main' into main

This commit is contained in:
BUAADreamer authored and GitHub committed 2024-04-25 20:02:50 +08:00
commit 68cdd9a020
8 files changed
+58 -23

No files matched your search

+1 -1
View File
@@ -169,7 +169,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,