Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
t0-0
commited on
Commit
•
2d8e1d1
1
Parent(s):
d7a1307
fix bug
Browse files- 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
|
43 |
precision = "bfloat16"
|
44 |
-
elif dtype
|
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(
|