simone-papicchio commited on
Commit
d952a61
·
1 Parent(s): f3d26e4

fix: removed unused models

Browse files
Files changed (1) hide show
  1. prediction.py +2 -3
prediction.py CHANGED
@@ -40,8 +40,7 @@ class ModelPrediction:
40
  self.model_name2pred_func = {
41
  "gpt-3.5": self._init_model_prediction("gpt-3.5"),
42
  "gpt-4o-mini": self._init_model_prediction("gpt-4o-mini"),
43
- "o1-mini": self._init_model_prediction("o1-mini"),
44
- "QwQ": self._init_model_prediction("QwQ"),
45
  "DeepSeek-R1-Distill-Llama-70B": self._init_model_prediction(
46
  "DeepSeek-R1-Distill-Llama-70B"
47
  ),
@@ -153,6 +152,6 @@ class ModelPrediction:
153
  model_name = "together_ai/meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo"
154
  else:
155
  raise ValueError("Model forbidden")
156
-
157
  return partial(predict_fun, model_name=model_name)
158
 
 
40
  self.model_name2pred_func = {
41
  "gpt-3.5": self._init_model_prediction("gpt-3.5"),
42
  "gpt-4o-mini": self._init_model_prediction("gpt-4o-mini"),
43
+ "llama-70": self._init_model_prediction("llama-70"),
 
44
  "DeepSeek-R1-Distill-Llama-70B": self._init_model_prediction(
45
  "DeepSeek-R1-Distill-Llama-70B"
46
  ),
 
152
  model_name = "together_ai/meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo"
153
  else:
154
  raise ValueError("Model forbidden")
155
+
156
  return partial(predict_fun, model_name=model_name)
157