diff --git a/Dockerfile b/Dockerfile
index 6b8e36da7..e8739d1da 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -50,4 +50,4 @@ EXPOSE 8188
 # Start as root so entrypoint can adjust ownership and drop privileges
 USER root
 ENTRYPOINT ["/entrypoint.sh"]
-CMD ["python", "main.py", "--listen", "0.0.0.0"]
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--use-sage-attention"]