Shaltiel commited on
Commit
1ba7e1c
·
1 Parent(s): 72bd0af

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