diff --git a/k8s/apps/comfyui/deployment.yaml b/k8s/apps/comfyui/deployment.yaml
index 83d2f8b..d483f70 100644
--- a/k8s/apps/comfyui/deployment.yaml
+++ b/k8s/apps/comfyui/deployment.yaml
@@ -35,17 +35,11 @@ spec:
               mountPath: /host.data
       containers:
         - name: comfyui
-          image: ghcr.io/ai-dock/comfyui:latest-cuda
+          image: runpod/comfyui:latest-5090
           imagePullPolicy: IfNotPresent
           env:
-            - name: COMFYUI_ARGS
-              value: "--listen 0.0.0.0"
-            - name: COMFYUI_PORT_LOCAL
+            - name: COMFYUI_PORT
               value: "8188"
-            - name: COMFYUI_PORT_HOST
-              value: "8189"
-            - name: COMFYUI_EXTRA_PIP_PACKAGES
-              value: "--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 --upgrade"
           ports:
             - containerPort: 8188
               name: http