diff --git a/setup.py b/setup.py index 1f99e39d..6f79f6b1 100644 --- a/setup.py +++ b/setup.py @@ -39,7 +39,7 @@ def get_cuda_version(): "opencv-python", ], extras_require={ - "[flash_attn]": [ + "flash_attn": [ "flash_attn>=2.6.3", ], },