support export size setting

This commit is contained in:
hiyouga
2023-11-26 18:34:09 +08:00
parent ff1c289229
commit 859a6ea942
5 changed files with 11 additions and 5 deletions

View File

@@ -149,6 +149,10 @@ class FinetuningArguments(FreezeArguments, LoraArguments, RLHFArguments):
default=None,
metadata={"help": "Path to the directory to save the exported model."}
)
export_size: Optional[int] = field(
default=1,
metadata={"help": "The file shard size (in GB) of the exported model."}
)
plot_loss: Optional[bool] = field(
default=False,
metadata={"help": "Whether to plot the training loss after fine-tuning or not."}