Hamza-Alobeidli commited on
Commit
cfa3bce
1 Parent(s): 64a51d7

Update src/display/utils.py

Browse files
Files changed (1) hide show
  1. src/display/utils.py +1 -1
src/display/utils.py CHANGED
@@ -102,7 +102,7 @@ class Precision(Enum):
102
  Unknown = ModelDetails("?")
103
 
104
  def from_str(precision):
105
- if precision in ["float32"]:
106
  return Precision.float32
107
  if precision in ["torch.float16", "float16"]:
108
  return Precision.float16
 
102
  Unknown = ModelDetails("?")
103
 
104
  def from_str(precision):
105
+ if precision in ["torch.float32", "float32"]:
106
  return Precision.float32
107
  if precision in ["torch.float16", "float16"]:
108
  return Precision.float16