update readme to point to direct link to runpod template, cleanup install instrucitons (#532)

* update readme to point to direct link to runpod template, cleanup install instrucitons

* default install flash-attn and auto-gptq now too

* update readme w flash-attn extra

* fix version in setup
This commit is contained in:
Wing Lian
2023-09-08 11:58:54 -04:00
committed by GitHub
parent 5e2d8a42d9
commit 34c0a86a11
5 changed files with 11 additions and 28 deletions

View File

@@ -15,9 +15,9 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
# If AXOLOTL_EXTRAS is set, append it in brackets
RUN cd axolotl && \
if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
pip install -e .[flash-attn,gptq,$AXOLOTL_EXTRAS]; \
pip install -e .[flash-attn,$AXOLOTL_EXTRAS]; \
else \
pip install -e .[flash-attn,gptq]; \
pip install -e .[flash-attn]; \
fi
# fix so that git fetch/pull from remote works