apiVersion: apps/v1
kind: Deployment
metadata:
  name: comfyui
  namespace: comfyui
  labels:
    app: comfyui
spec:
  replicas: 1
  selector:
    matchLabels:
      app: comfyui
  template:
    metadata:
      labels:
        app: comfyui
    spec:
      runtimeClassName: nvidia
      tolerations:
        - key: workload
          operator: Equal
          value: desktop
          effect: NoSchedule
      nodeSelector:
        kubernetes.io/hostname: uk-desktop.tail2fe2d.ts.net
      # Fix permissions mismatch usually happening when mapping host paths
      securityContext:
        runAsUser: 0
      containers:
        - name: comfyui
          image: ghcr.io/ai-dock/comfyui:latest-cuda
          imagePullPolicy: IfNotPresent
          env:
            - name: CLI_ARGS
              value: "--listen 0.0.0.0 --port 8188"
          ports:
            - containerPort: 8188
              name: http
              protocol: TCP
          resources:
            limits:
              nvidia.com/gpu: 1
          volumeMounts:
            - name: data
              # For ai-dock images, /workspace is the persistent user directory
              mountPath: /workspace
      volumes:
        - name: data
          persistentVolumeClaim:
            claimName: comfyui-data-pvc