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 |
# ----------------------------------
|