feat(llm-serving): scale ornith to 2 replicas instead of a dedicated grm GPU

reasoning keeps its 2 GPUs untouched. verifier's freed GPU goes to a second ornith replica instead of a standalone qwen-only pod -- both replicas load ornith:35b + qwen2.5:3b-instruct, k8s Service load-balances across them, so 2 concurrent implementer-style calls get independent instances.
This commit is contained in:
Story Crater Bot
2026-08-18 18:25:12 -07:00
parent 50d00ae350
commit 5d7b9acd43
4 changed files with 24 additions and 114 deletions
+8 -9
View File
@@ -7,14 +7,13 @@
# has a `path: k8s/apps/api` source) so all gateway config stays in one place. # has a `path: k8s/apps/api` source) so all gateway config stays in one place.
# #
# ── Model -> upstream map (verified live) ─────────────────────────────────── # ── Model -> upstream map (verified live) ───────────────────────────────────
# reasoning -> reasoning-predictor vLLM, DeepSeek-R1-Distill-32B # reasoning -> reasoning-predictor vLLM, DeepSeek-R1-Distill-32B, 2 replicas
# ornith:35b -> ornith-predictor Ollama # ornith:35b -> ornith-predictor Ollama, 2 replicas (retired verifier-
# qwen2.5:3b-instruct -> grm-predictor Ollama (dedicated GPU -- # qwen2.5:3b-instruct -> ornith-predictor Ollama predictor's vLLM PRM slot to get
# retired verifier-predictor's # the 2nd GPU) -- k8s Service load-balances
# vLLM PRM slot; verification/ # across both, each replica loads both
# judge traffic no longer # models, so 2 concurrent implementer-style
# contends with ornith:35b's # calls each land on an independent instance
# planner/implementer traffic)
# nomic-embed-text-v2 -> embeddings-predictor TEI # nomic-embed-text-v2 -> embeddings-predictor TEI
# bge-reranker-base -> reranker-predictor TEI # bge-reranker-base -> reranker-predictor TEI
# #
@@ -214,7 +213,7 @@ spec:
pathType: Prefix pathType: Prefix
backend: backend:
service: service:
name: grm-predictor name: ornith-predictor
port: port:
number: 80 number: 80
--- ---
-96
View File
@@ -1,96 +0,0 @@
apiVersion: serving.kserve.io/v1beta1
kind: InferenceService
metadata:
annotations:
serving.kserve.io/deploymentMode: RawDeployment
# Same Kong-timeout rationale as ornith.yaml: these configure the Service
# KServe generates, not the Ingress, and matter once OLLAMA_KEEP_ALIVE=-1
# stops covering a cold load after a pod restart.
konghq.com/connect-timeout: "10000"
konghq.com/read-timeout: "3600000"
konghq.com/write-timeout: "3600000"
labels:
app.kubernetes.io/name: llm-grm
app.kubernetes.io/part-of: llm-serving
name: grm
namespace: llm-serving
spec:
predictor:
containers:
- command:
- /bin/sh
- -c
- 'set -e
ollama serve &
SERVE_PID=$!
until ollama list >/dev/null 2>&1; do sleep 2; done
ollama pull qwen2.5:3b-instruct
ollama run qwen2.5:3b-instruct "ok" >/dev/null 2>&1 || true
wait $SERVE_PID
'
env:
- name: OLLAMA_HOST
value: 0.0.0.0:8080
- name: OLLAMA_MODELS
value: /mnt/models/ollama
- name: OLLAMA_CONTEXT_LENGTH
value: '32768'
- name: OLLAMA_KEEP_ALIVE
value: '-1'
- name: OLLAMA_NUM_PARALLEL
value: '1'
# Room for a second verification/reward model alongside qwen2.5:3b
# without a redeploy -- matches ornith.yaml's pattern, one dedicated
# GPU now free for it instead of contending with ornith:35b's.
- name: OLLAMA_MAX_LOADED_MODELS
value: '2'
image: ollama/ollama:0.32.9@sha256:1685741456770df6e3cceb2a945a5f75e020f658d1701509668d6f4688f1dd3f
name: kserve-container
ports:
- containerPort: 8080
protocol: TCP
readinessProbe:
exec:
command:
- /bin/sh
- -c
- ollama ps 2>/dev/null | grep -q qwen2.5
periodSeconds: 10
resources:
limits:
cpu: '16'
memory: 16Gi
nvidia.com/gpu: '1'
requests:
cpu: '8'
memory: 8Gi
nvidia.com/gpu: '1'
startupProbe:
exec:
command:
- /bin/sh
- -c
- ollama ps 2>/dev/null | grep -q qwen2.5
failureThreshold: 120
periodSeconds: 15
volumeMounts:
- mountPath: /mnt/models
name: models
deploymentStrategy:
type: Recreate
maxReplicas: 1
minReplicas: 1
nodeSelector:
kubernetes.io/hostname: worker-1
runtimeClassName: nvidia
volumes:
- name: models
persistentVolumeClaim:
claimName: llm-models
-1
View File
@@ -11,7 +11,6 @@ kind: Kustomization
# in tens of seconds, not a rolling update. # in tens of seconds, not a rolling update.
resources: resources:
- embeddings.yaml - embeddings.yaml
- grm.yaml
- ornith.yaml - ornith.yaml
- reasoning.yaml - reasoning.yaml
- reranker.yaml - reranker.yaml
+16 -8
View File
@@ -38,8 +38,12 @@ spec:
ollama pull ornith:35b ollama pull ornith:35b
ollama pull qwen2.5:3b-instruct
ollama run ornith:35b "ok" >/dev/null 2>&1 || true ollama run ornith:35b "ok" >/dev/null 2>&1 || true
ollama run qwen2.5:3b-instruct "ok" >/dev/null 2>&1 || true
wait $SERVE_PID wait $SERVE_PID
' '
@@ -54,11 +58,8 @@ spec:
value: '-1' value: '-1'
- name: OLLAMA_NUM_PARALLEL - name: OLLAMA_NUM_PARALLEL
value: '1' value: '1'
# qwen2.5:3b-instruct moved to its own dedicated GPU (llm-serving/grm.yaml)
# so verification/judge traffic no longer contends with ornith:35b's
# planner/implementer traffic on this one -- single model here now.
- name: OLLAMA_MAX_LOADED_MODELS - name: OLLAMA_MAX_LOADED_MODELS
value: '1' value: '2'
image: ollama/ollama:0.32.9@sha256:1685741456770df6e3cceb2a945a5f75e020f658d1701509668d6f4688f1dd3f image: ollama/ollama:0.32.9@sha256:1685741456770df6e3cceb2a945a5f75e020f658d1701509668d6f4688f1dd3f
name: kserve-container name: kserve-container
ports: ports:
@@ -69,7 +70,8 @@ spec:
command: command:
- /bin/sh - /bin/sh
- -c - -c
- ollama ps 2>/dev/null | grep -q ornith - ollama ps 2>/dev/null | grep -q ornith && ollama ps 2>/dev/null |
grep -q qwen2.5
periodSeconds: 10 periodSeconds: 10
resources: resources:
limits: limits:
@@ -85,7 +87,8 @@ spec:
command: command:
- /bin/sh - /bin/sh
- -c - -c
- ollama ps 2>/dev/null | grep -q ornith - ollama ps 2>/dev/null | grep -q ornith && ollama ps 2>/dev/null |
grep -q qwen2.5
failureThreshold: 120 failureThreshold: 120
periodSeconds: 15 periodSeconds: 15
volumeMounts: volumeMounts:
@@ -93,8 +96,13 @@ spec:
name: models name: models
deploymentStrategy: deploymentStrategy:
type: Recreate type: Recreate
maxReplicas: 1 # 2 replicas -- each its own GPU, each loading both ornith:35b and
minReplicas: 1 # qwen2.5:3b-instruct -- so 2 concurrent implementer-style calls each
# get an independent instance instead of contending on one, at the
# cost of judge/qwen traffic still sharing whichever replica an
# implementer call also lands on.
maxReplicas: 2
minReplicas: 2
nodeSelector: nodeSelector:
kubernetes.io/hostname: worker-1 kubernetes.io/hostname: worker-1
runtimeClassName: nvidia runtimeClassName: nvidia