winglian commited on
Commit
89134f2
1 Parent(s): 6086be8

make sure to install causal_conv1d in docker (#1459)

Browse files
Files changed (2) hide show
  1. cicd/Dockerfile.jinja +1 -0
  2. docker/Dockerfile +1 -0
cicd/Dockerfile.jinja CHANGED
@@ -22,6 +22,7 @@ RUN git fetch origin +$GITHUB_REF && \
22
  git checkout FETCH_HEAD
23
 
24
  # If AXOLOTL_EXTRAS is set, append it in brackets
 
25
  RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
26
  pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
27
  else \
 
22
  git checkout FETCH_HEAD
23
 
24
  # If AXOLOTL_EXTRAS is set, append it in brackets
25
+ RUN pip install causal_conv1d
26
  RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
27
  pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
28
  else \
docker/Dockerfile CHANGED
@@ -20,6 +20,7 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
20
  WORKDIR /workspace/axolotl
21
 
22
  # If AXOLOTL_EXTRAS is set, append it in brackets
 
23
  RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
24
  pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
25
  else \
 
20
  WORKDIR /workspace/axolotl
21
 
22
  # If AXOLOTL_EXTRAS is set, append it in brackets
23
+ RUN pip install causal_conv1d
24
  RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
25
  pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
26
  else \