Spaces:
Paused
Paused
Fixed endpoint model
Browse files
src/backend/run_eval_suite_lighteval.py
CHANGED
@@ -5,10 +5,12 @@ from datetime import datetime
|
|
5 |
from argparse import Namespace
|
6 |
|
7 |
from lighteval.main_accelerate import main, EnvConfig, create_model_config, load_model
|
8 |
-
|
9 |
from src.envs import RESULTS_REPO, CACHE_PATH, TOKEN
|
10 |
from src.backend.manage_requests import EvalRequest
|
11 |
|
|
|
|
|
12 |
logging.getLogger("openai").setLevel(logging.WARNING)
|
13 |
|
14 |
class DefaultNamespace(Namespace):
|
|
|
5 |
from argparse import Namespace
|
6 |
|
7 |
from lighteval.main_accelerate import main, EnvConfig, create_model_config, load_model
|
8 |
+
from lighteval.models.endpoint_model import InferenceEndpointModel
|
9 |
from src.envs import RESULTS_REPO, CACHE_PATH, TOKEN
|
10 |
from src.backend.manage_requests import EvalRequest
|
11 |
|
12 |
+
InferenceEndpointModel.add_special_tokens = lambda *args, **kwargs: None
|
13 |
+
|
14 |
logging.getLogger("openai").setLevel(logging.WARNING)
|
15 |
|
16 |
class DefaultNamespace(Namespace):
|