apiVersion: v1
kind: PersistentVolumeClaim
metadata:
  name: model-storage
spec:
  accessModes: ["ReadWriteMany"]
  resources:
    requests:
      storage: 500Gi
  storageClassName: shared-ssd-ord1

apiVersion: batch/v1
kind: Job
metadata:
  name: download-model
spec:
  template:
    spec:
      restartPolicy: Never
      containers:
        - name: downloader
          image: python:3.11-slim
          command: ["python3", "-c"]
          args:
            - |
              from huggingface_hub import snapshot_download
              snapshot_download("meta-llama/Llama-3.1-8B-Instruct", local_dir="/models/llama-8b")
          volumeMounts:
            - name: models
              mountPath: /models
          env:
            - name: HF_TOKEN
              valueFrom:
                secretKeyRef:
                  name: hf-token
                  key: token
      volumes:
        - name: models
          persistentVolumeClaim:
            claimName: model-storage