[bugfix]: unblock PyPI publish (flash-attn-cute direct dep) (#1436)
This commit is contained in:
+5
-1
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user