t0-0 commited on
Commit
2d8e1d1
1 Parent(s): d7a1307
Files changed (1) hide show
  1. src/submission/submit.py +2 -2
src/submission/submit.py CHANGED
@@ -39,9 +39,9 @@ def add_new_eval(
39
  dtype = config.torch_dtype
40
  if dtype == torch.float16:
41
  precision = "float16"
42
- elif dtype in torch.bfloat16:
43
  precision = "bfloat16"
44
- elif dtype in torch.float32:
45
  precision = "float32"
46
  else:
47
  return styled_error(
 
39
  dtype = config.torch_dtype
40
  if dtype == torch.float16:
41
  precision = "float16"
42
+ elif dtype == torch.bfloat16:
43
  precision = "bfloat16"
44
+ elif dtype == torch.float32:
45
  precision = "float32"
46
  else:
47
  return styled_error(