diff --git a/README.md b/README.md index eac297c97b..2bd8e8dc9a 100644 --- a/README.md +++ b/README.md @@ -205,7 +205,7 @@ pip install -e .\[flash_attn\] Installation using pip: ```shell -pip install trinity-rft==0.2.0 +pip install trinity-rft==0.2.1 # install flash-attn separately pip install flash-attn==2.8.0.post2 ``` diff --git a/README_zh.md b/README_zh.md index b65dae1006..7960de4555 100644 --- a/README_zh.md +++ b/README_zh.md @@ -204,7 +204,7 @@ pip install -e .\[flash_attn\] 使用 pip 安装: ```shell -pip install trinity-rft==0.2.0 +pip install trinity-rft==0.2.1 # flash-attn 需要单独安装 pip install flash-attn==2.8.0.post2 ``` diff --git a/docs/sphinx_doc/source/main.md b/docs/sphinx_doc/source/main.md index f5555d4a37..f83e9a812d 100644 --- a/docs/sphinx_doc/source/main.md +++ b/docs/sphinx_doc/source/main.md @@ -177,7 +177,7 @@ pip install -e .\[flash_attn\] Installation using pip: ```shell -pip install trinity-rft==0.2.0 +pip install trinity-rft==0.2.1 ``` Installation from docker: diff --git a/pyproject.toml b/pyproject.toml index 431410ad51..413917d923 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "trinity-rft" -version = "0.2.1.dev0" +version = "0.2.1" authors = [ {name="Trinity-RFT Team", email="trinity-rft@outlook.com"}, ] diff --git a/trinity/cli/launcher.py b/trinity/cli/launcher.py index e61a59d037..16c90e85d3 100644 --- a/trinity/cli/launcher.py +++ b/trinity/cli/launcher.py @@ -187,7 +187,7 @@ def run(config_path: str, dlc: bool = False, plugin_dir: str = None): config_path, ) - envs = {PLUGIN_DIRS_ENV_VAR: plugin_dir} + envs = {PLUGIN_DIRS_ENV_VAR: plugin_dir or ""} if dlc: from trinity.utils.dlc_utils import setup_ray_cluster