Add SWIFT coding agent probe experiment scripts

README.md

@@ -29,13 +29,37 @@ git submodule update --init --recursive
 ./scripts/setup_env.sh
 ```
 
-脚本会显式设置 B300 代理：
+脚本会显式设置 B300 代理和国内 PyPI 源：
 
 ```bash
 http://100.72.0.101:8888
+https://mirrors.aliyun.com/pypi/simple/
 ```
 
-Python 依赖安装在仓库内 `.venv`，不会写入系统 Python。
+Python 依赖安装在仓库内 `.venv`，不会写入系统 Python。安装逻辑是：
+
+1. 先在 `.venv` 内安装 `uv`。
+2. 用 SWIFT 官方推荐的 `uv pip install -e third_party/modelscope-swift --torch-backend=auto` 安装 SWIFT 源码依赖。
+3. 再安装本 probe repo。
+4. 默认额外补充 `deepspeed<0.19`、`liger-kernel`、`tensorboard`、`nvitop`。
+
+如果要安装 SWIFT 的全量 optional 依赖：
+
+```bash
+INSTALL_SWIFT_ALL=1 ./scripts/setup_env.sh
+```
+
+如果只想安装 SWIFT core，不补额外训练包：
+
+```bash
+INSTALL_TRAINING_EXTRAS=0 ./scripts/setup_env.sh
+```
+
+如果要追加其他包：
+
+```bash
+EXTRA_UV_PACKAGES="flash-attn==2.8.3" ./scripts/setup_env.sh
+```
 
 ## 下载数据集