Skip to content

Commit c35cd7d

Browse files
AnilAltinaygvisor-bot
authored andcommitted
Use the updated Docker image in ollama GKE tests.
The docker images used in the test is not getting updated since cl/613771002. PiperOrigin-RevId: 830639606
1 parent cd0901c commit c35cd7d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/kubernetes/benchmarks/ollama.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -902,8 +902,8 @@ const (
902902
ollamaPort = 11434
903903
ollamaPodName = "ollama-server"
904904
ollamaServiceName = "ollama-service"
905-
ollamaBenchImage = k8s.ImageRepoPrefix + "gpu/ollama/bench:latest"
906-
ollamaBenchClientImage = k8s.ImageRepoPrefix + "gpu/ollama/client:latest"
905+
ollamaBenchImage = k8s.ImageRepoPrefix + "gpu/ollama/bench_x86_64:latest"
906+
ollamaBenchClientImage = k8s.ImageRepoPrefix + "gpu/ollama/client_x86_64:latest"
907907
)
908908

909909
// newOllamaServerPod returns the pod spec for an ollama server.

0 commit comments

Comments
 (0)