abhisheksan commited on
Commit
5873aa2
·
1 Parent(s): acce523

Update requirements.txt to upgrade transformers and add vllm dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -9,7 +9,8 @@ python-multipart==0.0.9
9
 
10
  # For model handling
11
  torch==2.2.0
12
- transformers==4.37.2
 
13
  accelerate==0.27.2
14
 
15
  # Optional but recommended for performance
 
9
 
10
  # For model handling
11
  torch==2.2.0
12
+ transformers==4.43.1
13
+ vllm==0.5.3.post1
14
  accelerate==0.27.2
15
 
16
  # Optional but recommended for performance