Update deps (#390)

mirpo · Aug 23, 2024 · 09832a3 · 09832a3
1 parent 9e546f3
commit 09832a3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/templates/langchain/requirements.txt b/src/templates/langchain/requirements.txt
@@ -5,6 +5,7 @@ torch==2.4.0
 uvicorn==0.30.6
 langchain==0.2.14
 langchain-community==0.2.12
+accelerate==0.33.0
 # dev
 ruff==0.6.1
 pytest==8.3.2

diff --git a/src/templates/nlp/requirements.txt b/src/templates/nlp/requirements.txt
@@ -1,8 +1,9 @@
 fastapi==0.112.1
-transformers==4.42.4
+transformers[torch]==4.42.4
 pydantic-settings==2.4.0
 torch==2.4.0
 uvicorn==0.30.6
+accelerate==0.33.0
 # dev
 ruff==0.6.1
 pytest==8.3.2