Browse Source

fix flash attn install

tags/v0.3.9-rc1
haixuantao 11 months ago
parent
commit
ebcda89675
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      node-hub/dora-qwen2-5-vl/pyproject.toml

+ 5
- 5
node-hub/dora-qwen2-5-vl/pyproject.toml View File

@@ -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]


Loading…
Cancel
Save