Update dependencies and pin accelerate in testing

dottxt-ai · Sep 16, 2023 · 75cfe2d · 75cfe2d
1 parent 8b6c0b0
commit 75cfe2d
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,22 +35,19 @@ dependencies = [
    "scipy",
    "tenacity",
    "torch",
-   "accelerate",
 ]
 dynamic = ["version"]
 
 [project.optional-dependencies]
 test = [
     "diffusers",
     "pre-commit",
-    "pydantic>=2.0",
     "pytest",
     "pytest-cov",
     "transformers",
     "coverage[toml]>=5.1",
     "diff-cover",
-    "lark",
-    "interegular",
+    "accelerate<=0.22.0",
 ]
 
 [project.urls]