diff --git a/doc/en/install.md b/doc/en/install.md index 2799449..b3c42e4 100644 --- a/doc/en/install.md +++ b/doc/en/install.md @@ -7,7 +7,7 @@ The required packages and corresponding version are shown as follows: - MPFR == 4.1.0 - CUDA == 11.7 - Pytorch == 1.13.1+cu117 -- Transformers >= 4.25.1 +- Transformers >= 4.28.0 - Flash-Attention == v1.0.5 - Apex == 23.05 - GPU with Ampere or Hopper architecture (such as H100, A100) diff --git a/doc/install.md b/doc/install.md index c8eceeb..5af3a95 100644 --- a/doc/install.md +++ b/doc/install.md @@ -7,7 +7,7 @@ - MPFR == 4.1.0 - CUDA == 11.7 - Pytorch == 1.13.1+cu117 -- Transformers >= 4.25.1 +- Transformers >= 4.28.0 - Flash-Attention == v1.0.5 - Apex == 23.05 - Ampere或者Hopper架构的GPU (例如H100, A100)