Update requirements.txt

70f04859 · wanglch · 353a975f · 70f04859
Commit 70f04859 authored May 23, 2024 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

docker/requirements.txt docker/requirements.txt +3 -3

No files found.
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
-torch
-transformers
+torch>=2.1
+transformers==4.37.2
 accelerate
 colorama
 cpm_kernels
@@ -19,4 +19,4 @@ nltk
 uvicorn
 pydantic
 fastapi
-sse-starlette
\ No newline at end of file
+sse-starlette