Merge pull request #746 from containers/rhoai

Update vLLM containers
containers · Feb 6, 2025 · 1ab1908 · 1ab1908
2 parents 3477a13 + 8399024
commit 1ab1908
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ramalama/model.py b/ramalama/model.py
@@ -118,9 +118,9 @@ def _image(self, args):
 
         if args.runtime == "vllm":
             if gpu_type == "HIP_VISIBLE_DEVICES":
-                return "quay.io/modh/vllm:rhoai-2.17-rocm"
+                return "quay.io/modh/vllm:rhoai-2.18-rocm"
 
-            return "quay.io/modh/vllm:rhoai-2.17-cuda"
+            return "quay.io/modh/vllm:rhoai-2.18-cuda"
 
         split = version().split(".")
         vers = ".".join(split[:2])