Skip to content

Commit

Permalink
Removing flash_attn as it is too hard to auto-install
Browse files Browse the repository at this point in the history
  • Loading branch information
haixuanTao committed Feb 4, 2025
1 parent 1f6d109 commit f1e7c0e
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions node-hub/dora-qwen2-5-vl/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,16 +23,17 @@ dependencies = [
"accelerate>=1.3.0",
"transformers",
"setuptools>=65.0.0",
"flash-attn>=2.7.1; sys_platform != 'darwin'",
# "flash-attn>=2.7.1; sys_platform != 'darwin'",
]

[[tool.uv.dependency-metadata]]
name = "flash-attn"
version = "2.7.1"
requires-dist = ["torch"]
## Currently flash_attn is not supported as a pip install within uv.
# [[tool.uv.dependency-metadata]]
# name = "flash-attn"
# version = "2.7.1"
# requires = ["setuptools", "torch"]

[tool.uv]
no-build-isolation-package = ['flash-attn']
# [tool.uv]
# no-build-isolation-package = ['flash-attn']


[dependency-groups]
Expand Down

0 comments on commit f1e7c0e

Please sign in to comment.