From e65654acfd00a469b8a956737e62caba9c109ce4 Mon Sep 17 00:00:00 2001 From: "sungho.park" Date: Thu, 16 Mar 2023 16:02:39 +0000 Subject: [PATCH] add --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index b85c6c4..9b31bae 100644 --- a/train.py +++ b/train.py @@ -82,13 +82,13 @@ class ModelArguments: default=None, metadata={"help": "If training from scratch, pass a model type from the list: " + ", ".join(MODEL_TYPES)}, ) - lora_r: Optional[str] = field( + lora_r: Optional[float] = field( default=None, ) - lora_alpha: Optional[str] = field( + lora_alpha: Optional[float] = field( default=None, ) - lora_dropout: Optional[str] = field( + lora_dropout: Optional[float] = field( default=None, ) config_overrides: Optional[str] = field(