fix if elif seq2seq

huggingface · Jun 20, 2024 · fd6390c · fd6390c
1 parent 2cac237
commit fd6390c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -326,7 +326,7 @@ def launch_command(params):
                 "launch",
                 "--cpu",
             ]
-        if num_gpus == 1:
+        elif num_gpus == 1:
             cmd = [
                 "accelerate",
                 "launch",