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

updates

Browse Source
This commit is contained in:
Dan Saunders
2025-09-16 15:23:40 -04:00
parent 2acd3e1242
commit 02dc263338
2 changed files with 526 additions and 388 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pyproject.toml
View File

@@ -78,7 +78,7 @@ dependencies = [
# Platform-specific dependencies (Linux by default, excluded on macOS)
"triton>=3.0.0 ; sys_platform != 'darwin'",
"xformers>=0.0.23.post1 ; sys_platform != 'darwin'",
"xformers>=0.0.32.post1 ; sys_platform != 'darwin'",
"autoawq==0.2.7.post3 ; sys_platform != 'darwin'",
"liger-kernel==0.6.1 ; sys_platform != 'darwin'",
"torchao==0.13.0 ; sys_platform != 'darwin'",
@@ -204,7 +204,9 @@ dev-dependencies = [
]
find-links = [
"https://github.com/Dao-AILab/flash-attention/releases/expanded_assets/v2.8.3",
"https://github.com/ModelCloud/GPTQModel/releases",
]
prerelease = "allow"
# UV custom index for specific packages
[[tool.uv.index]]
@@ -220,7 +222,7 @@ flash-attn = [
"setuptools",
{ requirement = "torch", match-runtime = true },
]
auto-gptq = [
gptqmodel = [
{ requirement = "torch", match-runtime = true },
]
autoawq = ["torch"]