fix env
Browse files- environment.yaml +20 -0
environment.yaml
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: ats-models
|
| 2 |
+
channels:
|
| 3 |
+
- pytorch
|
| 4 |
+
- conda-forge
|
| 5 |
+
- defaults
|
| 6 |
+
dependencies:
|
| 7 |
+
- python=3.9
|
| 8 |
+
- pip
|
| 9 |
+
- pip:
|
| 10 |
+
- pytorch-lightning==1.9.3
|
| 11 |
+
- transformers
|
| 12 |
+
- torch
|
| 13 |
+
- tensorboardX
|
| 14 |
+
- rouge_score
|
| 15 |
+
- sacrebleu
|
| 16 |
+
- datasets
|
| 17 |
+
- sentencepiece
|
| 18 |
+
- openpyxl
|
| 19 |
+
- numpy==1.23.1
|
| 20 |
+
- bs4
|