From 773f3b568e770020224d652b68c235b0df180566 Mon Sep 17 00:00:00 2001
From: Vincent Chen <v.chen@databricks.com>
Date: Wed, 24 Jan 2024 14:27:00 -0800
Subject: [PATCH] 2.3.6 test

---
 docker/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index f92e307e1a..ee7331cb02 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -296,7 +296,7 @@ RUN if [[ -n "$CUDA_VERSION" ]] &&  [[ -z "${PYTORCH_NIGHTLY_URL}" ]]; then \
 RUN if [ -n "$CUDA_VERSION" ] ; then \
         pip${PYTHON_VERSION} install --upgrade --no-cache-dir ninja==1.11.1 && \
         pip${PYTHON_VERSION} install --upgrade --no-cache-dir --force-reinstall packaging==22.0 && \
-        pip${PYTHON_VERSION} install --no-cache-dir flash-attn==2.5.0; \
+        pip${PYTHON_VERSION} install --no-cache-dir flash-attn==2.3.6; \
     fi
 
 ###############