diff --git a/flaml/nlp/utils.py b/flaml/nlp/utils.py index 489d00b96..357dbe0ae 100644 --- a/flaml/nlp/utils.py +++ b/flaml/nlp/utils.py @@ -609,7 +609,7 @@ class HFArgs: def __post_init__(self): if self.task in NLG_TASKS: - HFArgs.model_path = "t5-small" + self.model_path = "t5-small" @staticmethod def load_args_from_console(): diff --git a/notebook/automl_nlp.ipynb b/notebook/automl_nlp.ipynb index c74d89c9d..7c255ceb9 100644 --- a/notebook/automl_nlp.ipynb +++ b/notebook/automl_nlp.ipynb @@ -1398,7 +1398,7 @@ "\n", "automl_settings = {\n", " \"time_budget\": 500, # setting the time budget\n", - " \"task\": \"summarization\", # setting the task as multiplechoice-classification\n", + " \"task\": \"summarization\", # setting the task as summarization\n", " \"hf_args\": {\n", " \"output_dir\": \"data/output/\", # setting the output directory\n", " \"ckpt_per_epoch\": 1, # setting the number of checkoints per epoch\n",