Spaces:
Sleeping
Sleeping
Commit
·
d952a61
1
Parent(s):
f3d26e4
fix: removed unused models
Browse files- 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 |
-
"
|
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 |
|