Auto init with mistral
Browse files- parapluie.py +1 -1
- requirements.txt +1 -3
parapluie.py
CHANGED
|
@@ -126,7 +126,7 @@ class Parapluie(evaluate.Metric):
|
|
| 126 |
def _compute(self, sources, hypotheses):
|
| 127 |
if self.scorer is None:
|
| 128 |
print("Init hasn't been done ! Auto init")
|
| 129 |
-
self.init(model="
|
| 130 |
print("Loading Mistral Done")
|
| 131 |
scores = []
|
| 132 |
for i in range(len(sources)):
|
|
|
|
| 126 |
def _compute(self, sources, hypotheses):
|
| 127 |
if self.scorer is None:
|
| 128 |
print("Init hasn't been done ! Auto init")
|
| 129 |
+
self.init(model="mistralai/Mistral-7B-Instruct-v0.2", device="cpu")
|
| 130 |
print("Loading Mistral Done")
|
| 131 |
scores = []
|
| 132 |
for i in range(len(sources)):
|
requirements.txt
CHANGED
|
@@ -1,6 +1,4 @@
|
|
| 1 |
git+https://github.com/huggingface/evaluate@main
|
| 2 |
transformers>=4.38.2
|
| 3 |
torch>=2.2.1
|
| 4 |
-
numpy>=1.26.4
|
| 5 |
-
einops
|
| 6 |
-
sentencepiece
|
|
|
|
| 1 |
git+https://github.com/huggingface/evaluate@main
|
| 2 |
transformers>=4.38.2
|
| 3 |
torch>=2.2.1
|
| 4 |
+
numpy>=1.26.4
|
|
|
|
|
|