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

dep

Browse Source
This commit is contained in:
Dan Saunders
2025-09-15 17:13:45 -04:00
parent 0437c1a4ba
commit 2acd3e1242
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pyproject.toml
View File

@@ -81,7 +81,7 @@ dependencies = [
"xformers>=0.0.23.post1 ; sys_platform != 'darwin'",
"autoawq==0.2.7.post3 ; sys_platform != 'darwin'",
"liger-kernel==0.6.1 ; sys_platform != 'darwin'",
"torchao==0.12.0 ; sys_platform != 'darwin'",
"torchao==0.13.0 ; sys_platform != 'darwin'",
"bitsandbytes==0.47.0 ; sys_platform != 'darwin'",
"flash-attn==2.8.3 ; sys_platform == 'linux'",
"deepspeed>=0.17.5 ; sys_platform != 'darwin'",
@@ -89,7 +89,6 @@ dependencies = [
]
[project.optional-dependencies]
# Flash attention context parallelism
ring-flash-attn = [
"flash-attn==2.8.3 ; sys_platform == 'linux'",
"ring-flash-attn>=0.1.7",