Add pytorch

Harrolee · Apr 14, 2024 · c252b6f · c252b6f
1 parent f1ebee7
commit c252b6f
Show file tree

Hide file tree

Showing 2 changed files with 290 additions and 1 deletion.
diff --git a/api/poetry.lock b/api/poetry.lock
diff --git a/api/pyproject.toml b/api/pyproject.toml
@@ -21,6 +21,7 @@ requests = "^2.31.0"
 flask-login = "^0.6.3"
 langchain-openai = "^0.0.8"
 transformers = "^4.39.3"
+torch = "^2.2.2"
 
 [build-system]
 requires = ["poetry-core"]