qlemesle commited on
Commit
53e33f3
·
1 Parent(s): b1732b6

Auto init with mistral

Browse files
Files changed (2) hide show
  1. parapluie.py +1 -1
  2. 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="internlm/internlm2-chat-1_8b", device="cpu")
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