Update README.md

This commit is contained in:
figerhaowang 2025-09-15 20:48:07 +08:00 committed by GitHub
parent f55e098602
commit 150e38c945
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 11 deletions

View File

@ -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:
``` ```