fix llm predict (#6870)

PaddlePaddle · Aug 30, 2023 · 3655108 · 3655108
1 parent 2aa58d8
commit 3655108
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm/predictor.py b/llm/predictor.py
@@ -71,7 +71,7 @@ class PredictorArgument:
 class ModelArgument:
     gpt: bool = field(default=False, metadata={"help": "GPTForCausalLM"})
     ernie: bool = field(default=False, metadata={"help": "Ernie35ForCausalLM"})
-    data_file: None = field(default=None, metadata={"help": "data file directory"})
+    data_file: str = field(default=None, metadata={"help": "data file directory"})
     output_file: str = field(default="output.json", metadata={"help": "predict result file directory"})