fix export format

This commit is contained in:
hiyouga
2023-12-28 18:40:46 +08:00
parent 5431be42f9
commit e165354fac
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