Harmonize ollama Docker image version.

The benchmark image was updated to 0.5.1, so this updates the other images to the same version. PiperOrigin-RevId: 706801311
google · Dec 16, 2024 · 3f058a8 · 3f058a8
1 parent 232c17c
commit 3f058a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/images/gpu/ollama/Dockerfile.x86_64 b/images/gpu/ollama/Dockerfile.x86_64
@@ -1,5 +1,5 @@
 # https://hub.docker.com/r/ollama/ollama
-FROM ollama/ollama:0.1.26
+FROM ollama/ollama:0.5.1
 
 ENV PATH=$PATH:/usr/local/nvidia/bin:/bin/nvidia/bin
 ENV OLLAMA_ORIGINS=*

diff --git a/images/gpu/ollama/singlemodel/Dockerfile.x86_64 b/images/gpu/ollama/singlemodel/Dockerfile.x86_64
@@ -1,5 +1,5 @@
 # https://hub.docker.com/r/ollama/ollama
-FROM ollama/ollama:0.1.26
+FROM ollama/ollama:0.5.1
 
 ENV PATH=$PATH:/usr/local/nvidia/bin:/bin/nvidia/bin
 ENV OLLAMA_ORIGINS=*