build: fixing torch import

2 years ago · 01911685f6
parent 8651b93fce
commit 01911685f6
1 changed files with 1 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -23,6 +23,7 @@ redis = "^4.3.1"
 dill = "^0.3.5"
 Flask = "^2.1.2"
 transformers = "^4.19.2"
+torch = "^1.8"
 requests = "^2.27.1"
 tqdm = "^4.64.0"
 types-redis = "^4.2.6"