Setting upper bound for accelerate to avoid torch dependency conflicts

sdv-dev · Jan 31, 2025 · 361bd7d · 361bd7d
1 parent f992575
commit 361bd7d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -64,7 +64,11 @@ sdgym = { main = 'sdgym.cli.__main__:main' }
 
 [project.optional-dependencies]
 dask = ['dask', 'distributed']
-realtabformer = ['realtabformer>=0.2.1', 'transformers<4.46']
+realtabformer = [
+    'realtabformer>=0.2.1',
+    'transformers<4.46',
+    "accelerate<1.3.0;python_version<'3.10'",
+]
 test = [
     'sdgym[realtabformer]',
     'pytest>=6.2.5',