* Fix 'no such option: -U' error (#696)

huggingface · Jul 8, 2024 · 8d33a3d · 8d33a3d
1 parent 37fd5b0
commit 8d33a3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autotrain/cli/run_setup.py b/src/autotrain/cli/run_setup.py
@@ -37,7 +37,7 @@ def run(self):
         if self.colab:
             cmd = "pip install -U xformers==0.0.24"
         else:
-            cmd = "pip uninstall -U xformers"
+            cmd = "pip uninstall -y xformers"
         cmd = cmd.split()
         pipe = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         logger.info("Installing latest xformers")