diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py index c3d92de4d9..0f5d0ea224 100644 --- a/src/autotrain/__init__.py +++ b/src/autotrain/__init__.py @@ -41,7 +41,7 @@ warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub") logger = Logger().get_logger() -__version__ = "0.8.24.dev0" +__version__ = "0.8.24" def is_colab(): diff --git a/src/autotrain/app/ui_routes.py b/src/autotrain/app/ui_routes.py index fcd861d754..ee1ebbf80a 100644 --- a/src/autotrain/app/ui_routes.py +++ b/src/autotrain/app/ui_routes.py @@ -284,6 +284,11 @@ "type": "number", "label": "Max doc stride", }, + "distributed_backend": { + "type": "dropdown", + "label": "Distributed backend", + "options": ["ddp", "deepspeed"], + }, }