diff --git a/requirements.txt b/requirements.txt index 48e8185..f11f601 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ sentencepiece tensorboard tensorboardX torch -vllm +vllm==0.6.2 vllm-flash-attn nltk fraction