chore: added config sentence transformers
Browse files- config.json +3 -2
- environment.yaml +2 -1
- requirements.txt +1 -0
config.json
CHANGED
|
@@ -31,7 +31,7 @@
|
|
| 31 |
}
|
| 32 |
],
|
| 33 |
"dependencies": {
|
| 34 |
-
"python_version": "3.
|
| 35 |
"packages": [
|
| 36 |
"torch==2.0.1",
|
| 37 |
"torchvision==0.15.2",
|
|
@@ -43,7 +43,8 @@
|
|
| 43 |
"uvicorn==0.23.2",
|
| 44 |
"numpy==1.24.4",
|
| 45 |
"opencv-python==4.8.0.76",
|
| 46 |
-
"scipy==1.
|
|
|
|
| 47 |
]
|
| 48 |
}
|
| 49 |
}
|
|
|
|
| 31 |
}
|
| 32 |
],
|
| 33 |
"dependencies": {
|
| 34 |
+
"python_version": "3.10",
|
| 35 |
"packages": [
|
| 36 |
"torch==2.0.1",
|
| 37 |
"torchvision==0.15.2",
|
|
|
|
| 43 |
"uvicorn==0.23.2",
|
| 44 |
"numpy==1.24.4",
|
| 45 |
"opencv-python==4.8.0.76",
|
| 46 |
+
"scipy==1.12.0",
|
| 47 |
+
"sentence-transformers==2.2.2"
|
| 48 |
]
|
| 49 |
}
|
| 50 |
}
|
environment.yaml
CHANGED
|
@@ -163,6 +163,7 @@ dependencies:
|
|
| 163 |
- torchmetrics==1.1.0
|
| 164 |
- tqdm==4.66.1
|
| 165 |
- transformers==4.32.0
|
|
|
|
| 166 |
- triton==2.0.0
|
| 167 |
- tzdata==2023.3
|
| 168 |
- urllib3==1.26.16
|
|
@@ -186,4 +187,4 @@ dependencies:
|
|
| 186 |
- torchvision==0.15.2
|
| 187 |
- xformers==0.0.22
|
| 188 |
- diffusers==0.21.4
|
| 189 |
-
prefix: /home/tiger/miniconda3/envs/manimate
|
|
|
|
| 163 |
- torchmetrics==1.1.0
|
| 164 |
- tqdm==4.66.1
|
| 165 |
- transformers==4.32.0
|
| 166 |
+
- sentence-transformers==2.2.2
|
| 167 |
- triton==2.0.0
|
| 168 |
- tzdata==2023.3
|
| 169 |
- urllib3==1.26.16
|
|
|
|
| 187 |
- torchvision==0.15.2
|
| 188 |
- xformers==0.0.22
|
| 189 |
- diffusers==0.21.4
|
| 190 |
+
prefix: /home/tiger/miniconda3/envs/manimate
|
requirements.txt
CHANGED
|
@@ -102,6 +102,7 @@ toolz==0.12.0
|
|
| 102 |
torchmetrics==1.1.0
|
| 103 |
tqdm==4.66.1
|
| 104 |
transformers==4.32.0
|
|
|
|
| 105 |
triton==2.0.0
|
| 106 |
tzdata==2023.3
|
| 107 |
urllib3==1.26.16
|
|
|
|
| 102 |
torchmetrics==1.1.0
|
| 103 |
tqdm==4.66.1
|
| 104 |
transformers==4.32.0
|
| 105 |
+
sentence-transformers==2.2.2
|
| 106 |
triton==2.0.0
|
| 107 |
tzdata==2023.3
|
| 108 |
urllib3==1.26.16
|