From 89134f2143cd3325802813eb97cd05c783932201 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Fri, 29 Mar 2024 16:43:25 -0400
Subject: [PATCH] make sure to install causal_conv1d in docker (#1459)

---
 cicd/Dockerfile.jinja | 1 +
 docker/Dockerfile     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja
index 19d7a09de5..ce03e08c2f 100644
--- a/cicd/Dockerfile.jinja
+++ b/cicd/Dockerfile.jinja
@@ -22,6 +22,7 @@ RUN git fetch origin +$GITHUB_REF && \
     git checkout FETCH_HEAD
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
+RUN pip install causal_conv1d
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 84046c6522..ae3fe89ae4 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -20,6 +20,7 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 WORKDIR /workspace/axolotl
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
+RUN pip install causal_conv1d
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \