example(LLM): change chatglm3 generation_config field type for comman…

…d line arguments (#3128)
star-whale · Jan 17, 2024 · 55955c1 · 55955c1
1 parent ea8c736
commit 55955c1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/example/llm-finetune/models/chatglm3/finetune.py b/example/llm-finetune/models/chatglm3/finetune.py
@@ -72,6 +72,10 @@ class TrainArguments(Seq2SeqTrainingArguments):
     remove_unused_columns: bool = dataclasses.field(
         default=False, metadata={"help": "remove unused columns"}
     )
+    # Currently, we don't support `Optional[Union[str, Path, GenerationConfig]]` for the command line.
+    generation_config: t.Optional[str] = dataclasses.field(
+        default=None, metadata={"help": "generation config json file path"}
+    )
 
 
 @argument((ModelArguments, DatasetArguments, TrainArguments))