Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Clémentine
commited on
Commit
•
6902167
1
Parent(s):
18abd06
changed default instance to cpu just in case
Browse files- main_backend_lighteval.py +4 -1
- 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 |
-
|
|
|
|
|
|
|
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 = "
|
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 |
# ----------------------------------
|