Spaces:
Paused
Paused
Adjusted caps and params
Browse files
main_backend_lighteval.py
CHANGED
@@ -61,12 +61,14 @@ def run_auto_eval():
|
|
61 |
raise ValueError("Couldn't detect number of params, please make sure the metadata is available")
|
62 |
elif params_size < 4:
|
63 |
instance_size, instance_type, cap = "x1", "nvidia-t4", 40
|
64 |
-
elif params_size <
|
65 |
instance_size, instance_type, cap = "x1", "nvidia-a10g", 40
|
66 |
elif params_size < 30:
|
67 |
instance_size, instance_type, cap = "x4", "nvidia-a10g", 20
|
68 |
elif params_size < 45:
|
69 |
instance_size, instance_type, cap = "x2", "nvidia-a100", 5
|
|
|
|
|
70 |
else:
|
71 |
set_eval_request(
|
72 |
api=API,
|
|
|
61 |
raise ValueError("Couldn't detect number of params, please make sure the metadata is available")
|
62 |
elif params_size < 4:
|
63 |
instance_size, instance_type, cap = "x1", "nvidia-t4", 40
|
64 |
+
elif params_size < 8:
|
65 |
instance_size, instance_type, cap = "x1", "nvidia-a10g", 40
|
66 |
elif params_size < 30:
|
67 |
instance_size, instance_type, cap = "x4", "nvidia-a10g", 20
|
68 |
elif params_size < 45:
|
69 |
instance_size, instance_type, cap = "x2", "nvidia-a100", 5
|
70 |
+
elif params_size < 80:
|
71 |
+
instance_size, instance_type, cap = "x4", "nvidia-a100", 5
|
72 |
else:
|
73 |
set_eval_request(
|
74 |
api=API,
|