diff --git a/Dockerfile b/Dockerfile
index 2b16d8a..73848c5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,7 +9,15 @@ WORKDIR /app/ComfyUI
 FROM pytorch/pytorch:2.9.0-cuda13.0-cudnn9-runtime
 WORKDIR /app
 COPY --from=builder /app/ComfyUI /app/ComfyUI
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    git \
+    curl \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
 WORKDIR /app/ComfyUI
+RUN curl -1sLf 'https://dl.cloudsmith.io/public/task/task/setup.deb.sh' | bash
+RUN curl -sSL https://install.python-poetry.org | python -
 RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 8188
 CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "8188"]
diff --git a/compose.yaml b/compose.yaml
index e39cd02..6046eb3 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -7,12 +7,16 @@ services:
       - CUDA_VISIBLE_DEVICES=0
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+      - POETRY_VENV_IN_PROJECT=false
+      - POETRY_NO_INTERACTION=1
+      - POETRY_CACHE_DIR=/tmp/poetry_cache
     volumes:
       - ./docker/models/checkpoints:/app/ComfyUI/models/checkpoints
       - ./docker/models/loras:/app/ComfyUI/models/loras
       - ./docker/models/vae:/app/ComfyUI/models/vae
       - ./docker/output:/app/ComfyUI/output
       - ./docker/workflows:/app/ComfyUI/user/default/workflows
+      - .:/app/ComfyUI/custom_nodes/ComfyUI-Diffusers
     ports:
       - "8188:8188"
     restart: unless-stopped