update readme

xdit-project · Dec 26, 2024 · fe7434e · fe7434e
1 parent e2dbc97
commit fe7434e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -224,7 +224,7 @@ pip install "xfuser[diffusers,flash]"  # With both diffusers and flash attention
 ```
 pip install -e .
 # Or optionally, with diffusers
-pip install -e ".[diffusers]"
+pip install -e ".[diffusers,flash-attn]"
 ```
 
 Note that we use two self-maintained packages:

diff --git a/setup.py b/setup.py
@@ -45,7 +45,7 @@ def get_cuda_version():
             "diffusers": [
                 "diffusers==0.31.0",  # NOTE: diffusers>=0.32.0.dev is necessary for CogVideoX and Flux
             ],
-            "flash": [
+            "flash-attn": [
                 "flash-attn>=2.6.0",
             ]
         },