Bump vllm

Inpher · Apr 9, 2024 · bcc6a0f · bcc6a0f
1 parent e3e414e
commit bcc6a0f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/backend/python/common-env/transformers/transformers-nvidia.yml b/backend/python/common-env/transformers/transformers-nvidia.yml
@@ -116,8 +116,8 @@ dependencies:
       - sudachipy
       - sudachidict_core
       - vocos
-      - vllm==0.3.2
-      - transformers>=4.38.2  # Updated Version
+      - vllm==0.4.0.post1
+      - transformers>=4.39.1
       - transformers_stream_generator==0.0.5
       - xformers==0.0.23.post1  
 prefix: /opt/conda/envs/transformers
diff --git a/backend/python/common-env/transformers/transformers-rocm.yml b/backend/python/common-env/transformers/transformers-rocm.yml
@@ -104,8 +104,8 @@ dependencies:
       - sudachipy
       - sudachidict_core
       - vocos
-      - vllm==0.3.2
-      - transformers>=4.38.2  # Updated Version
+      - vllm==0.4.0.post1
+      - transformers>=4.39.1
       - transformers_stream_generator==0.0.5
       - xformers==0.0.23.post1
 prefix: /opt/conda/envs/transformers
diff --git a/backend/python/common-env/transformers/transformers.yml b/backend/python/common-env/transformers/transformers.yml
@@ -108,8 +108,8 @@ dependencies:
       - sudachipy
       - sudachidict_core
       - vocos
-      - vllm==0.3.2
-      - transformers>=4.38.2  # Updated Version
+      - vllm==0.4.0.post1
+      - transformers>=4.39.1
       - transformers_stream_generator==0.0.5
       - xformers==0.0.23.post1  
 prefix: /opt/conda/envs/transformers