llama-3.2-nemoretriever-500m-rerank-v2

Get API Key and Install the NIM Operator

Install the NVIDIA GPU Operator

Copy Code
helm repo add nvidia https://helm.ngc.nvidia.com/nvidia \
   && helm repo update

helm install nim-operator nvidia/k8s-nim-operator --create-namespace -n nim-operator

Create a ImagePull Secrets

Copy Code
kubectl create ns nim-service

kubectl create secret -n nim-service docker-registry ngc-secret \
    --docker-server=nvcr.io \
    --docker-username='$oauthtoken' \
    --docker-password=<PASTE_API_KEY_HERE>

kubectl create secret -n nim-service generic ngc-api-secret \
    --from-literal=NGC_API_KEY=<PASTE_API_KEY_HERE>

Create a NIM Service

Ensure that a default StorageClass exists in the cluster. If none is present, create an appropriate StorageClass before proceeding.

NOTE:

• Select model-size based on the model and GPU type as described here.
• For example, change the nvidia.com/gpu: 1 based on the model and number of GPU requirements

Copy Code
apiVersion: apps.nvidia.com/v1alpha1
kind: NIMService
metadata:
  name: llama-32-nemoretriever-500m-rerank-v2
  namespace: nim-service
spec:
  image:
    repository: nvcr.io/nim/nvidia/llama-3.2-nemoretriever-500m-rerank-v2
    tag: latest
    pullPolicy: IfNotPresent
    pullSecrets:
      - ngc-secret
  authSecret: ngc-api-secret
  storage:
    pvc:
      create: true
      size: "model-size"
      volumeAccessMode: "ReadWriteOnce"
  replicas: 1
  resources:
    limits:
      nvidia.com/gpu: 1
  expose:
    service:
      type: ClusterIP
      port: 8000

Test the Deployed NIM

Copy Code
kubectl run --rm -it -n default curl --image=curlimages/curl:latest -- ash

Copy Code
curl -X "POST" \
 'http://llama-32-nemoretriever-500m-rerank-v2.nim-service:8000/v1/ranking' \
  -H 'Accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
        "model": "nvidia/llama-3.2-nemoretriever-500m-rerank-v2",
        "query": {"text": "which way did the traveler go?"},
        "passages": [
          {"text": "two roads diverged in a yellow wood, and sorry i could not travel both and be one traveler, long i stood and looked down one as far as i could to where it bent in the undergrowth;"},
          {"text": "then took the other, as just as fair, and having perhaps the better claim because it was grassy and wanted wear, though as for that the passing there had worn them really about the same,"},
          {"text": "and both that morning equally lay in leaves no step had trodden black. oh, i marked the first for another day! yet knowing how way leads on to way i doubted if i should ever come back."},
          {"text": "i shall be telling this with a sigh somewhere ages and ages hense: two roads diverged in a wood, and i, i took the one less traveled by, and that has made all the difference."}
        ],
        "truncate": "END"
      }'

For more details on getting started with this NIM, visit the NVIDIA NIM Operator Docs.