Spaces:
Paused
Paused
Added t4s back in for small models
Browse files
main_backend_lighteval.py
CHANGED
@@ -55,8 +55,8 @@ def run_auto_eval():
|
|
55 |
# For GPU
|
56 |
if not eval_request or eval_request.params < 0:
|
57 |
raise ValueError("Couldn't detect number of params, please make sure the metadata is available")
|
58 |
-
|
59 |
-
|
60 |
elif eval_request.params < 9:
|
61 |
instance_size, instance_type, cap = "x1", "nvidia-a10g", 40
|
62 |
elif eval_request.params < 30:
|
|
|
55 |
# For GPU
|
56 |
if not eval_request or eval_request.params < 0:
|
57 |
raise ValueError("Couldn't detect number of params, please make sure the metadata is available")
|
58 |
+
elif eval_request.params < 4:
|
59 |
+
instance_size, instance_type, cap = "x1", "nvidia-t4", 40
|
60 |
elif eval_request.params < 9:
|
61 |
instance_size, instance_type, cap = "x1", "nvidia-a10g", 40
|
62 |
elif eval_request.params < 30:
|