diff --git a/k8s/apps/comfyui/deployment.yaml b/k8s/apps/comfyui/deployment.yaml
index ddde40e..8c2fba5 100644
--- a/k8s/apps/comfyui/deployment.yaml
+++ b/k8s/apps/comfyui/deployment.yaml
@@ -16,6 +16,11 @@ spec:
         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