fix export format

Former-commit-id: 7c82bd396b9e6ff395850ad544d95cbf1b7557cd
This commit is contained in:
hiyouga
2023-12-28 18:40:46 +08:00
parent 53d7c5109f
commit af3f5b6e16
3 changed files with 29 additions and 1 deletions

View File

@@ -102,6 +102,10 @@ class ModelArguments:
default=1024,
metadata={"help": "The maximum length of the model inputs used for quantization."}
)
export_lecacy_format: Optional[bool] = field(
default=False,
metadata={"help": "Whether or not to save the `.bin` files instead of `.safetensors`."}
)
def __post_init__(self):
self.compute_dtype = None