bump version + colab fix

huggingface · Oct 23, 2023 · 503a0c9 · 503a0c9
1 parent 676d310
commit 503a0c9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -30,4 +30,4 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="tensorflow")
 
 
-__version__ = "0.6.41.dev0"
+__version__ = "0.6.42.dev0"
diff --git a/src/autotrain/cli/run_setup.py b/src/autotrain/cli/run_setup.py
@@ -60,9 +60,9 @@ def run(self):
         logger.info("Successfully installed latest trl")
 
         if self.colab:
-            cmd = "pip install -U xformers"
+            cmd = "pip install -U xformers==0.0.22"
         else:
-            cmd = "pip uninstall -U xformers==0.0.22"
+            cmd = "pip uninstall -U xformers"
         pipe = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         logger.info("Installing latest xformers")
         _, _ = pipe.communicate()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -30,4 +30,4 @@
		warnings.filterwarnings("ignore", category=UserWarning, module="tensorflow")


		__version__ = "0.6.41.dev0"
		__version__ = "0.6.42.dev0"