Omartificial-Intelligence-Space
commited on
Commit
•
1193d13
1
Parent(s):
55fde54
update utils
Browse files- src/display/utils.py +1 -1
src/display/utils.py
CHANGED
@@ -199,4 +199,4 @@ class Precision(Enum):
|
|
199 |
return Precision.float16
|
200 |
if precision_str in ["torch.bfloat16", "bfloat16"]:
|
201 |
return Precision.bfloat16
|
202 |
-
return Precision.Unknown
|
|
|
199 |
return Precision.float16
|
200 |
if precision_str in ["torch.bfloat16", "bfloat16"]:
|
201 |
return Precision.bfloat16
|
202 |
+
return Precision.Unknown
|