From f01acb67f634ff936066618bc1d3c19c3296d1e3 Mon Sep 17 00:00:00 2001 From: Susan Xueqing Liu Date: Tue, 9 May 2023 20:48:22 -0400 Subject: [PATCH] update model of text summarization (#1030) --- test/nlp/test_autohf_summarization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)