Clémentine commited on
Commit
6902167
1 Parent(s): 18abd06

changed default instance to cpu just in case

Browse files
Files changed (2) hide show
  1. main_backend_lighteval.py +4 -1
  2. src/envs.py +1 -1
main_backend_lighteval.py CHANGED
@@ -62,7 +62,10 @@ def run_auto_eval():
62
 
63
  # This needs to be done
64
  #instance_size, instance_type = get_instance_for_model(eval_request)
65
- instance_size, instance_type = "small", "g4dn.xlarge"
 
 
 
66
 
67
  run_evaluation(
68
  eval_request=eval_request,
 
62
 
63
  # This needs to be done
64
  #instance_size, instance_type = get_instance_for_model(eval_request)
65
+ # For GPU
66
+ # instance_size, instance_type = "small", "g4dn.xlarge"
67
+ # For CPU
68
+ instance_size, instance_type = "medium", "c6i"
69
 
70
  run_evaluation(
71
  eval_request=eval_request,
src/envs.py CHANGED
@@ -13,7 +13,7 @@ DEVICE = "cpu" # "cuda:0" if you add compute, for harness evaluations
13
  LIMIT = 20 # !!!! Should be None for actual evaluations!!!
14
 
15
  # For lighteval evaluations
16
- ACCELERATOR = "gpu"
17
  REGION = "us-east-1"
18
  VENDOR = "aws"
19
  # ----------------------------------
 
13
  LIMIT = 20 # !!!! Should be None for actual evaluations!!!
14
 
15
  # For lighteval evaluations
16
+ ACCELERATOR = "cpu"
17
  REGION = "us-east-1"
18
  VENDOR = "aws"
19
  # ----------------------------------