fix bugs

aJupyter · Aug 12, 2024 · f418585 · f418585
1 parent 29e9974
commit f418585
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train_args/dpo/dpo_config.py b/train_args/dpo/dpo_config.py
@@ -51,7 +51,7 @@ class TrainArgument(DPOConfig):
     fp16: bool = field(default=False, metadata={"help": "Whether to use fp16 (mixed) precision instead of 32-bit"})
 
     # Deepspeed训练相关参数，不使用时设置为default=None
-    deepspeed: Optional[str] = field(default='./train_args/deepspeed_config/ds_config_zero2.json',
+    deepspeed: Optional[str] = field(default=None,
                                      metadata={"help": "启用Deepspeed时需要的config文件"})
     # ---------------------------------------------------------------------------------------------------------------------
     # 上面参数是常规TrainingArguments设置，下面参数则是dpo配置参数。下面为DPOConfig默认配置。

diff --git a/train_args/sft/lora_qlora/base.py b/train_args/sft/lora_qlora/base.py
@@ -42,5 +42,5 @@ class TrainArgument(TrainingArguments):
     fp16: bool = field(default=False, metadata={"help": "Whether to use fp16 (mixed) precision instead of 32-bit"})
 
     # Deepspeed训练相关参数，不使用时设置为default=None
-    deepspeed: Optional[str] = field(default='./train_args/deepspeed_config/ds_config_zero2.json',
+    deepspeed: Optional[str] = field(default=None,
                                      metadata={"help": "启用Deepspeed时需要的config文件"})