Spaces:
Paused
Paused
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()
|