apiVersion: apps/v1
kind: StatefulSet
metadata:
  name: comfyui
  namespace: comfyui
spec:
  replicas: 1
  selector:
    matchLabels:
      app: comfyui
  serviceName: comfyui
  template:
    metadata:
      labels:
        app: comfyui
    spec:
      restartPolicy: Always
      terminationGracePeriodSeconds: 30
      nodeSelector:
        kubernetes.io/hostname: PLACEHOLDER
      containers:
        - name: comfyui
          image: ghcr.io/ai-dock/comfyui:latest
          ports:
            - containerPort: 8188
          env:
            - name: COMFYUI_FLAGS
              value: "--listen 0.0.0.0 --port 8188 --use-xpu"
            - name: AUTO_UPDATE
              value: "false"
          securityContext:
            privileged: true
          volumeMounts:
            - name: data
              mountPath: /config
            - name: dri
              mountPath: /dev/dri
      volumes:
        - name: data
          persistentVolumeClaim:
            claimName: comfyui-pvc
        - name: dri
          hostPath:
            path: /dev/dri