tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 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

13
.github/workflows/multi-gpu-e2e.yml vendored
View File

@@ -21,10 +21,17 @@ jobs:
pytorch: 2.3.1
axolotl_extras:
num_gpus: 2
- cuda: 121
cuda_version: 12.1.1
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.3.1
pytorch: 2.4.1
axolotl_extras:
num_gpus: 2
nightly_build: "true"
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.5.0
axolotl_extras:
num_gpus: 2
nightly_build: "true"