diff --git a/requirements.txt b/requirements.txt index 802d6f2..c1103d7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,6 @@ Flask-Caching==2.0.2 requests~=2.31.0 tiktoken~=0.5.0 gevent~=23.9.0.post1 -torch~=2.0.1 PyMySQL~=1.1.0 simplejson~=3.19.1 websockets~=11.0.3 @@ -14,4 +13,5 @@ urllib3~=2.0.4 flask-sock==0.6.0 gunicorn==21.2.0 redis==5.0.1 -ujson==5.8.0 \ No newline at end of file +ujson==5.8.0 +vllm