Update README.md
This commit is contained in:
parent
f55e098602
commit
150e38c945
20
README.md
20
README.md
|
|
@ -41,19 +41,17 @@ This repository contains the implementation of REPST, a framework for spatio-tem
|
||||||
## Requirements
|
## Requirements
|
||||||
Use python 3.11 from MiniConda
|
Use python 3.11 from MiniConda
|
||||||
|
|
||||||
- torch==2.2.2
|
- torch==2.0.1
|
||||||
- accelerate==0.28.0
|
- accelerate==0.28.0
|
||||||
- einops==0.7.0
|
- einops==0.6.0
|
||||||
- matplotlib==3.7.0
|
- matplotlib==3.7.0
|
||||||
- numpy==1.23.5
|
- numpy==1.24.4
|
||||||
- pandas==1.5.3
|
- pandas==2.1.4
|
||||||
- scikit_learn==1.2.2
|
- scikit_learn==1.3.2
|
||||||
- scipy==1.12.0
|
- scipy==1.11.4
|
||||||
- tqdm==4.65.0
|
- tqdm==4.66.1
|
||||||
- peft==0.4.0
|
- transformers==4.36.2
|
||||||
- transformers==4.31.0
|
|
||||||
- deepspeed==0.14.0
|
|
||||||
- sentencepiece==0.2.0
|
|
||||||
|
|
||||||
To install all dependencies:
|
To install all dependencies:
|
||||||
```
|
```
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue