Update src/display/utils.py

src/display/utils.py

@@ -102,7 +102,7 @@ class Precision(Enum):
     Unknown = ModelDetails("?")
 
     def from_str(precision):
-        if precision in ["float32"]:
+        if precision in ["torch.float32", "float32"]:
             return Precision.float32
         if precision in ["torch.float16", "float16"]:
             return Precision.float16