Shaltiel commited on
Commit
2c38fe4
1 Parent(s): 4eba5d7

Fixed envs

Browse files
Files changed (1) hide show
  1. src/envs.py +1 -1
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 = "cuda:0"
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 = "gpu"
17
  REGION = "us-east-1"
18
  VENDOR = "aws"
19
  # ----------------------------------