update model of text summarization (microsoft#1030)

TeBoxx · May 10, 2023 · f01acb6 · f01acb6
1 parent 59e882e
commit f01acb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)