feat(poetry): install accelerate python library

walledai · Jun 21, 2024 · 136118d · 136118d
1 parent 3bd6c9c
commit 136118d
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,6 +28,7 @@ transformers = "^4.40.1"
 datasets = "^2.19.0"
 pydantic = "^2.7.1"
 python-dotenv = "^1.0.1"
+accelerate = "^0.31.0"
 
 [tool.poetry.group.dev.dependencies]  
 pytest = "^7.4"