Skip to content

Commit 9df3b31

Browse files
samhita-allaAtharva1723
authored andcommitted
pin ollama version (flyteorg#2997)
Signed-off-by: Samhita Alla <[email protected]> Signed-off-by: Atharva <[email protected]>
1 parent 2b298cf commit 9df3b31

File tree

1 file changed

+1
-1
lines changed
  • plugins/flytekit-inference/flytekitplugins/inference/ollama

1 file changed

+1
-1
lines changed

plugins/flytekit-inference/flytekitplugins/inference/ollama/serve.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def __init__(self, *args, **kwargs):
164164
name=container_name,
165165
image="python:3.11-slim",
166166
command=["/bin/sh", "-c"],
167-
args=[f"pip install requests && pip install ollama && {command}"],
167+
args=[f"pip install requests && pip install ollama==0.3.3 && {command}"],
168168
resources=V1ResourceRequirements(
169169
requests={
170170
"cpu": self._model_cpu,

0 commit comments

Comments
 (0)