From 550d1cfe9bbd503df05c1392968360b56eb928f8 Mon Sep 17 00:00:00 2001 From: Dhruv Thakur <80753201+dhruvthakur2000@users.noreply.github.com> Date: Sat, 10 Feb 2024 13:02:57 +0530 Subject: [PATCH] Update AutoML-NLP.md (#1239) * Update AutoML-NLP.md #834 * more space --------- Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Chi Wang --- website/docs/Examples/AutoML-NLP.md | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/website/docs/Examples/AutoML-NLP.md b/website/docs/Examples/AutoML-NLP.md index 2896ff89d5..2619fda56a 100644 --- a/website/docs/Examples/AutoML-NLP.md +++ b/website/docs/Examples/AutoML-NLP.md @@ -27,10 +27,9 @@ automl_settings = { "time_budget": 100, "task": "seq-classification", "fit_kwargs_by_estimator": { - "transformer": - { - "output_dir": "data/output/" # if model_path is not set, the default model is facebook/muppet-roberta-base: https://huggingface.co/facebook/muppet-roberta-base - } + "transformer": { + "output_dir": "data/output/" # if model_path is not set, the default model is facebook/muppet-roberta-base: https://huggingface.co/facebook/muppet-roberta-base + } }, # setting the huggingface arguments: output directory "gpu_per_trial": 1, # set to 0 if no GPU is available }