From fe7434eb784feb3b427d38feb53c8391fb495831 Mon Sep 17 00:00:00 2001
From: feifeibear <fangjiarui123@gmail.com>
Date: Thu, 26 Dec 2024 19:42:00 +0800
Subject: [PATCH] update readme

---
 README.md | 2 +-
 setup.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 2c0c1c6..91b054f 100644
--- 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
index 1a85782..7985bff 100644
--- 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",
             ]
         },