Update src/display/utils.py

src/display/utils.py

@@ -120,25 +120,32 @@ class WeightType(Enum):
 
 
 class Precision(Enum):
+    Other = ModelDetails("Other")
+    float64 = ModelDetails("float64")
+    float32 = ModelDetails("float32")
     float16 = ModelDetails("float16")
     bfloat16 = ModelDetails("bfloat16")
     qt_8bit = ModelDetails("8bit")
     qt_4bit = ModelDetails("4bit")
     qt_GPTQ = ModelDetails("GPTQ")
-    Unknown = ModelDetails("?")
+    #Unknown = ModelDetails("?")
 
     def from_str(precision):
-        if precision in ["torch.float16", "float16"]:
+        if precision in ["torch.float64", "torch.double" ,"float64"]:
+            return Precision.float64
+        if precision in ["torch.float32", "torch.float" ,"float32"]:
+            return Precision.tfloat32
+        if precision in ["torch.float16", "torch.half", "float16"]:
             return Precision.float16
         if precision in ["torch.bfloat16", "bfloat16"]:
             return Precision.bfloat16
-        if precision in ["8bit"]:
+        if precision in ["8bit", "int8"]:
             return Precision.qt_8bit
-        if precision in ["4bit"]:
+        if precision in ["4bit", "int4"]:
             return Precision.qt_4bit
         if precision in ["GPTQ", "None"]:
             return Precision.qt_GPTQ
-        return Precision.Unknown
+        return Precision.Other
 
 
 # Column selection