diff --git a/README.md b/README.md
index c1a56ee6..7c21084a 100644
--- a/README.md
+++ b/README.md
@@ -141,18 +141,26 @@ Here are the benchmark results for Pixart-Alpha using the 20-step DPM solver as
 
 <h2 id="QuickStart">🚀 QuickStart</h2>
 
-1. Install yunchang for sequence parallel.
+### 1. Install from pip
+
+```
+pip install xfuser
+```
+
+### 2. Install from source 
+
+#### 2.1 Install yunchang for sequence parallel.
 
 Install yunchang from [feifeibear/long-context-attention](https://github.com/feifeibear/long-context-attention).
 Please note that it has a dependency on flash attention and specific GPU model requirements. We recommend installing yunchang from the source code rather than using `pip install yunchang==0.2.0`.
 
-2. Install xDiT
+#### 2.2 Install xDiT
 
 ```
 python setup.py install
 ```
 
-3. Usage
+### 2. Usage
 
 We provide examples demonstrating how to run models with xDiT in the [./examples/](./examples/) directory. 
 You can easily modify the model type, model directory, and parallel options in the [examples/run.sh](examples/run.sh) within the script to run some already supported DiT models.
diff --git a/dist/pipefusion-0.2-py3.11.egg b/dist/pipefusion-0.2-py3.11.egg
deleted file mode 100644
index c0553737..00000000
Binary files a/dist/pipefusion-0.2-py3.11.egg and /dev/null differ
diff --git a/examples/run.sh b/examples/run.sh
index d05f2dec..e05a207d 100644
--- a/examples/run.sh
+++ b/examples/run.sh
@@ -35,7 +35,7 @@ elif [ "$MODEL_TYPE" = "Sd3" ]; then
     export INFERENCE_STEP=20
 elif [ "$MODEL_TYPE" = "Flux" ]; then
     export SCRIPT=flux_example.py
-    export MODEL_ID="/cfs/dit/FLUX.1-schnell"
+    export MODEL_ID="/mnt/models/SD/FLUX.1-schnell"
     export INFERENCE_STEP=4
     # Flux does not apply cfg
     export CFG_ARGS=""
@@ -44,7 +44,7 @@ else
     exit 1
 fi
 
-export PYTHONAPTH=$PWD:$PYTHONPATH
+# export PYTHONAPTH=$PWD:$PYTHONPATH
 
 mkdir -p ./results
 
diff --git a/setup.py b/setup.py
index e0b27fbe..3e859124 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@
             "diffusers>=0.30.0",
             "transformers",
             "sentencepiece",
-            "accelerate",
+            "accelerate==0.33.0",
             "beautifulsoup4",
             "distvae",
             "ftfy",
diff --git a/xfuser/__version__.py b/xfuser/__version__.py
index a4e2017f..09888577 100644
--- a/xfuser/__version__.py
+++ b/xfuser/__version__.py
@@ -1 +1 @@
-__version__ = "0.1"
+__version__ = "0.2"
diff --git a/xfuser/modules/__init__.py b/xfuser/modules/__init__.py
new file mode 100644
index 00000000..e69de29b