[bugfix]: unblock PyPI publish (flash-attn-cute direct dep) (#1436)

This commit is contained in:
William Lin
2026-06-05 10:07:12 -07:00
committed by GitHub
parent efcc245c2e
commit 041e5e9029
+5 -1
View File
@@ -106,6 +106,7 @@ torchaudio = [
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
imagebind = { git = "https://github.com/facebookresearch/ImageBind.git", rev = "53680b02d7e37b19b124fa37bae4b6c98c38f5be" }
flash-attn-cute = { git = "https://github.com/XOR-op/flash-attention.git", branch = "fa4-compile", subdirectory = "flash_attn/cute" }
[[tool.uv.index]]
name = "pytorch-cpu"
@@ -175,7 +176,10 @@ streaming = [
dreamverse = [
"uvicorn[standard]>=0.41.0",
"cerebras-cloud-sdk",
"flash-attn-cute @ git+https://github.com/XOR-op/flash-attention.git@fa4-compile#subdirectory=flash_attn/cute",
# PyPI forbids direct URL deps in published metadata; pin the fork via
# [tool.uv.sources] above (same pattern as imagebind) so the wheel stays
# publishable while uv workspace installs still get the fork.
"flash-attn-cute",
"flashinfer-python",
"openai>=1.40",
]