Shaltiel commited on
Commit
d3573c0
·
1 Parent(s): c55e8f9

Adjusted caps and params

Browse files
Files changed (1) hide show
  1. main_backend_lighteval.py +3 -1
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 < 9:
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,