diff --git a/README.md b/README.md index 8de76a354..881602e8f 100644 --- a/README.md +++ b/README.md @@ -95,8 +95,6 @@ conda install mpi4py bash install.sh ``` -You may try `pip install packaging torch` before `bash install.sh` if the installation process freezed due to limited RAM. - ### Prepare Dataset Please refer to our [doc](https://github.com/OptimalScale/LMFlow/blob/main/docs/dataset.md). diff --git a/requirements.txt b/requirements.txt index 4f8ea964c..97d20a2e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ +packaging numpy==1.24.2 datasets==2.14.6 tokenizers>=0.13.3