|
|
|
@@ -22,13 +22,13 @@ dependencies = [ |
|
|
|
"peft == 0.13.2", |
|
|
|
"accelerate>=1.3.0", |
|
|
|
"transformers", |
|
|
|
"flash-attn>=v2.7.1; sys_platform != 'darwin'", |
|
|
|
"flash-attn; sys_platform != 'darwin'", |
|
|
|
] |
|
|
|
|
|
|
|
[tool.uv] |
|
|
|
no-build-isolation-package = ['flash-attn'] |
|
|
|
|
|
|
|
# flash_attn = "^2.6.1" # Install using: pip install -U flash-attn --no-build-isolation |
|
|
|
[[tool.uv.dependency-metadata]] |
|
|
|
name = "flash-attn" |
|
|
|
version = ">=2.7.1" |
|
|
|
requires-dist = ["torch"] |
|
|
|
|
|
|
|
|
|
|
|
[dependency-groups] |
|
|
|
|