diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py index c9a8cb9427..eb8b4e18c6 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.7.126" +__version__ = "0.7.127" def is_colab(): diff --git a/src/autotrain/commands.py b/src/autotrain/commands.py index d1adfadd4e..84893a50f2 100644 --- a/src/autotrain/commands.py +++ b/src/autotrain/commands.py @@ -326,7 +326,7 @@ def launch_command(params): "launch", "--cpu", ] - if num_gpus == 1: + elif num_gpus == 1: cmd = [ "accelerate", "launch",