diff --git a/pyproject.toml b/pyproject.toml index 4b0ca40f2..621d8d721 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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", ]