Merge pull request #57 from NickDatLe/upgrade_to_python311

Updated requirements and README.md to use Python 3.11 for better model efficiency
This commit is contained in:
Ming Jin 2024-04-23 11:33:57 +10:00 committed by GitHub
commit 6b9c40a2aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 8 deletions

View File

@ -64,18 +64,21 @@ Notably, we show that time series analysis (e.g., forecasting) can be cast as ye
</p>
## Requirements
- accelerate==0.20.3
Use python 3.11 from MiniConda
- torch==2.2.2
- accelerate==0.28.0
- einops==0.7.0
- matplotlib==3.7.0
- numpy==1.23.5
- pandas==1.5.3
- scikit_learn==1.2.2
- scipy==1.5.4
- torch==2.0.1
- scipy==1.12.0
- tqdm==4.65.0
- peft==0.4.0
- transformers==4.31.0
- deepspeed==0.13.0
- deepspeed==0.14.0
- sentencepiece==0.2.0
To install all dependencies:
```

View File

@ -1,12 +1,13 @@
accelerate==0.20.3
torch==2.2.2
accelerate==0.28.0
einops==0.7.0
matplotlib==3.7.0
numpy==1.23.5
pandas==1.5.3
scikit_learn==1.2.2
scipy==1.5.4
torch==2.0.1
scipy==1.12.0
tqdm==4.65.0
peft==0.4.0
transformers==4.31.0
deepspeed==0.13.0
deepspeed==0.14.0
sentencepiece==0.2.0