diff --git a/README.md b/README.md index 6fc13a4..1473a16 100644 --- a/README.md +++ b/README.md @@ -109,9 +109,9 @@ Recommended Runtime Environment: | cuda | | cuda12.8/13.0 | | | torch | >=2.0 | 2.8.0/2.11.0 | | | transformer-engine | >=2.3 | 2.14.1 | | -| apex | | 0.1 | | +| apex | | 0.1 | Optional | | megatron-core | >=0.15,<0.18 | 0.17.0 | | -| flash-attn | | 2.8.3/3.0.0b1 | | +| flash-attn | | 2.8.3/3.0.0b1 | Optional | | transformers | >=4.33 | 4.57.6/5.8.1 | | | modelscope | >=1.23 | | | | peft | >=0.11,<0.20 | | LoRA | diff --git a/README_zh.md b/README_zh.md index 25069ca..5f170e7 100644 --- a/README_zh.md +++ b/README_zh.md @@ -107,9 +107,9 @@ uv pip install -e . --torch-backend=auto | cuda | | cuda12.8/13.0 | | | torch | >=2.0 | 2.8.0/2.11.0 | | | transformer-engine | >=2.3 | 2.14.1 | | -| apex | | 0.1 | | +| apex | | 0.1 | 可选 | | megatron-core | >=0.15,<0.18 | 0.17.0 | | -| flash-attn | | 2.8.3/3.0.0b1 | | +| flash-attn | | 2.8.3/3.0.0b1 | 可选 | | transformers | >=4.33 | 4.57.6/5.8.1 | | | modelscope | >=1.23 | | | | peft | >=0.11,<0.20 | | LoRA |