diff --git a/RecLM-gen/requirements.txt b/RecLM-gen/requirements.txt index 10c34dd..9c833c8 100644 --- a/RecLM-gen/requirements.txt +++ b/RecLM-gen/requirements.txt @@ -12,7 +12,7 @@ sanic==23.12.1 torch==2.1.2 tqdm==4.66.1 transformers==4.36.2 -vllm==0.2.7 +vllm==0.7.2 flash-attn==2.3.6 Levenshtein~=0.23.0 unirec~=0.0.1a5 \ No newline at end of file