diff --git a/Dockerfile b/Dockerfile
index 7061fe61a..e4ef06124 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:24.03-py3
+FROM nvcr.io/nvidia/pytorch:24.12-py3
 
 ARG TZ="America/Los_Angeles"
 
@@ -6,13 +6,14 @@ ENV PYTORCH_CUDA_ALLOC_CONF="backend:cudaMallocAsync"
 ENV UV_COMPILE_BYTECODE=1
 ENV UV_NO_CACHE=1
 ENV UV_SYSTEM_PYTHON=1
+ENV UV_BREAK_SYSTEM_PACKAGES=1
 ENV PIP_DISABLE_PIP_VERSION_CHECK=1
 ENV PIP_NO_CACHE_DIR=1
 ENV DEBIAN_FRONTEND="noninteractive"
 # mitigates
 # RuntimeError: Failed to import transformers.generation.utils because of the following error (look up to see its traceback):
 # numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject
-RUN echo "numpy<2" > numpy-override.txt
+RUN echo "numpy<2" >> numpy-override.txt
 
 # mitigates https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
 # mitigates AttributeError: module 'cv2.dnn' has no attribute 'DictValue' \
@@ -20,10 +21,10 @@ RUN echo "numpy<2" > numpy-override.txt
 RUN pip install uv && uv --version && \
     apt-get update && apt-get install --no-install-recommends ffmpeg libsm6 libxext6 -y && \
     uv pip uninstall --system $(pip list --format=freeze | grep opencv) && \
-    rm -rf /usr/local/lib/python3.10/dist-packages/cv2/ && \
+    rm -rf /usr/local/lib/python3.12/dist-packages/cv2/ && \
     uv pip install wheel && \
     uv pip install --no-build-isolation opencv-python-headless && \
-    uv pip install --no-build-isolation --overrides=numpy-override.txt "comfyui@git+https://github.com/hiddenswitch/ComfyUI.git" && \
+    uv pip install --overrides=numpy-override.txt "comfyui[attention]@git+https://github.com/hiddenswitch/ComfyUI.git" && \
     rm -rf /var/lib/apt/lists/*
 
 WORKDIR /workspace