diff --git a/services/invoke/Dockerfile b/services/invoke/Dockerfile
index 21c13ec..15282a3 100644
--- a/services/invoke/Dockerfile
+++ b/services/invoke/Dockerfile
@@ -26,7 +26,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
   pip install -e .
 
 
-ARG BRANCH=main SHA=f3b2e02921927d9317255b1c3811f47bd40a2bf9
+ARG BRANCH=main SHA=19cdd5a99b5402fbef9240c67ef8351d5309450d
 RUN --mount=type=cache,target=/root/.cache/pip \
   git fetch && \
   git reset --hard && \
@@ -47,7 +47,7 @@ ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR
 EXPOSE 7860
 
 ENTRYPOINT ["/docker/entrypoint.sh"]
-CMD invokeai --web --host 0.0.0.0 --port 7860 --root_dir ${ROOT} --config ${CONFIG_DIR}/models.yaml \
-  --outdir /output/invoke --embedding_directory /data/embeddings/ --lora_directory /data/models/Lora \
-  --no-nsfw_checker --no-safety_checker ${CLI_ARGS}
+CMD invokeai --web --host 0.0.0.0 --port 7860 --root ${ROOT} --conf_path ${CONFIG_DIR}/models.yaml \
+  --outdir /output/invoke --embedding_dir /data/embeddings/ --lora_dir /data/models/Lora --controlnet_dir /data/models/ControlNet \
+  --no-nsfw_checker --no-nsfw_checker ${CLI_ARGS}