make sure to install causal_conv1d in docker (#1459)
Browse files- cicd/Dockerfile.jinja +1 -0
- 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 \
|