support export size setting

Former-commit-id: 859a6ea942
This commit is contained in:
hiyouga
2023-11-26 18:34:09 +08:00
parent 5f2943dc84
commit b015ac35d8
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."}