qlemesle commited on
Commit
0bf76c7
·
1 Parent(s): a34a6a0

Auto init with interlm2

Browse files
Files changed (1) hide show
  1. parapluie.py +1 -1
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="google/gemma-2-2b-it", 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="internlm/internlm2-chat-1_8b", device="cpu")
130
  print("Loading Mistral Done")
131
  scores = []
132
  for i in range(len(sources)):