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 |
|