add option to disable version check

Former-commit-id: fd769cb2de696aee3c5e882237e16eace6a9d675
2024-02-10 22:31:23 +08:00
parent 62b6a7971a
commit 5f83860aa1
4 changed files with 47 additions and 26 deletions
--- a/src/llmtuner/hparams/finetuning_args.py
+++ b/src/llmtuner/hparams/finetuning_args.py
@@ -132,6 +132,9 @@ class FinetuningArguments(FreezeArguments, LoraArguments, RLHFArguments):
    finetuning_type: Optional[Literal["lora", "freeze", "full"]] = field(
        default="lora", metadata={"help": "Which fine-tuning method to use."}
    )
+    disable_version_checking: Optional[bool] = field(
+        default=False, metadata={"help": "Whether or not to disable version checking."}
+    )
    plot_loss: Optional[bool] = field(
        default=False, metadata={"help": "Whether or not to save the training loss curves."}
    )