Shaltiel commited on
Commit
59447e3
·
1 Parent(s): f100cf0

Added t4s back in for small models

Browse files
Files changed (1) hide show
  1. main_backend_lighteval.py +2 -2
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
- # elif eval_request.params < 4:
59
- # instance_size, instance_type, cap = "x1", "nvidia-a10g", 20
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: