Merge pull request caikit#147 from Maxusmusti/dependency-update

Update Accelerate dependency to fix import
Ssukriti · Aug 29, 2023 · 0592555 · 0592555
2 parents 6f555e8 + c83787d
commit 0592555
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,7 @@ dependencies = [
     "caikit[runtime-grpc,runtime-http]>=0.16.0,<0.18.0",
     "caikit-tgis-backend>=0.1.16,<0.2.0",
     # TODO: loosen dependencies
-    "accelerate>=0.18.0",
+    "accelerate>=0.21.0",
     "datasets>=2.4.0",
     "huggingface-hub",
     "numpy>=1.22.4",