tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

first pass at pytorch 2.5.0 support (#1982)

Browse Source 
* first pass at pytorch 2.5.0 support

* attempt to install causal_conv1d with mamba

* gracefully handle missing xformers

* fix import

* fix incorrect version, add 2.5.0

* increase tests timeout
This commit is contained in:
Wing Lian
2024-10-21 11:00:45 -04:00
committed by GitHub
parent 67f744dc8c
commit e12a2130e9
12 changed files with 120 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
cicd/Dockerfile.jinja
View File

@@ -23,7 +23,6 @@ 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 [ "$NIGHTLY_BUILD" = "true" ] ; then \
sed -i 's#^transformers.*#transformers @ git+https://github.com/huggingface/transformers.git@main#' requirements.txt; \
sed -i 's#^peft.*#peft @ git+https://github.com/huggingface/peft.git@main#' requirements.txt; \