Shaltiel commited on
Commit
73adc1c
1 Parent(s): 0e63ee0

Fixed cleanup with reuse existing

Browse files
src/backend/run_eval_suite_lighteval.py CHANGED
@@ -69,7 +69,7 @@ def run_evaluation(eval_request: EvalRequest, task_names: str, batch_size: int,
69
  import traceback
70
  traceback.print_exception(ex)
71
  env_config = EnvConfig(token=TOKEN, cache_dir=args.cache_dir)
72
- args.reuse_existing = True
73
  model_config = create_model_config(args=args, accelerator=accelerator)
74
  model, _ = load_model(config=model_config, env_config=env_config)
75
  model.cleanup()
 
69
  import traceback
70
  traceback.print_exception(ex)
71
  env_config = EnvConfig(token=TOKEN, cache_dir=args.cache_dir)
72
+ args.model_config['model']['base_params']['reuse_existing'] = True
73
  model_config = create_model_config(args=args, accelerator=accelerator)
74
  model, _ = load_model(config=model_config, env_config=env_config)
75
  model.cleanup()