Adding Precision

app.py

@@ -23,6 +23,23 @@ class ModelDetails:
     display_name: str = ""
     symbol: str = "" # emoji
 
+class Precision(Enum):
+    float16 = ModelDetails("float16")
+    bfloat16 = ModelDetails("bfloat16")
+    float32 = ModelDetails("float32")
+    bfloat32 = ModelDetails("bfloat32")
+    Unknown = ModelDetails("?")
+
+    def from_str(precision):
+        if precision in ["torch.float16", "float16"]:
+            return Precision.float16
+        if precision in ["torch.bfloat16", "bfloat16"]:
+            return Precision.bfloat16
+        if precision in ["torch.bfloat32", "bfloat32"]:
+            return Precision.bfloat32
+        if precision in ["torch.float32", "float32"]:
+            return Precision.float32
+            
 class ModelType(Enum):
     PT = ModelDetails(name="pretrained", symbol="🟢")
     FT = ModelDetails(name="fine-tuned", symbol="🔶")