diff --git a/test/nlp/test_autohf_summarization.py b/test/nlp/test_autohf_summarization.py index e54a42bdb..9d2687dae 100644 --- a/test/nlp/test_autohf_summarization.py +++ b/test/nlp/test_autohf_summarization.py @@ -22,7 +22,7 @@ def test_summarization(): automl_settings["task"] = "summarization" automl_settings["metric"] = "rouge1" automl_settings["time_budget"] = 2 * automl_settings["time_budget"] - automl_settings["fit_kwargs_by_estimator"]["transformer"]["model_path"] = "patrickvonplaten/t5-tiny-random" + automl_settings["fit_kwargs_by_estimator"]["transformer"]["model_path"] = "google/flan-t5-small" try: automl.fit(X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings)