add

README.md

@@ -137,9 +137,16 @@ mim install "mmcv>=2.0.0"
 **Step 4**: Install other dependencies.
 
 ```shell
-pip install -U transformers wandb einops pycocotools shapely scipy terminaltables
+pip install -U transformers wandb einops pycocotools shapely scipy terminaltables importlib
 ```
 
+**Step 5**: [Optional] Install DeepSpeed.
+
+If you want to use DeepSpeed to train the model, you need to install DeepSpeed. The installation method of DeepSpeed can refer to the [DeepSpeed official document](https://github.com/microsoft/DeepSpeed).
+Note: The support for DeepSpeed under the Windows system is not perfect yet, we recommend that you use DeepSpeed under the Linux system.
+
+
+
 </details>
 
 ### Install RSPrompter

README_zh-CN.md

@@ -140,7 +140,13 @@ mim install "mmcv>=2.0.0"
 ```shell
 pip install -U transformers wandb einops pycocotools shapely scipy terminaltables importlib
 ```
-[可选]安装deepspeed 
+
+**步骤 5**：[可选] 安装 DeepSpeed。
+
+如果您想使用 DeepSpeed 训练模型，您需要安装 DeepSpeed。DeepSpeed 的安装方法可以参考 [DeepSpeed 官方文档](https://github.com/microsoft/DeepSpeed)。
+
+注解：Windows 系统下对 DeepSpeed 的支持尚未完善，我们建议您在 Linux 系统下使用 DeepSpeed。
+
 
 </details>