Update src/display/utils.py

src/display/utils.py

@@ -240,16 +240,15 @@ class WeightDtype(Enum):
     int2 = ModelDetails("int2")
     int3 = ModelDetails("int3")    
     int4 = ModelDetails("int4")
+    int8 = ModelDetails("int8")    
     nf4 = ModelDetails("nf4")
     fp4 = ModelDetails("fp4")
-    fp16 = ModelDetails("float16")
+    f16 = ModelDetails("float16")
     bf16 = ModelDetails("bfloat16")
-    fp32 = ModelDetails("float32")
+    f32 = ModelDetails("float32")
 
     Unknown = ModelDetails("?")
 
-
-
     def from_str(weight_dtype):
         if weight_dtype in ["int2"]:
             return WeightDtype.int2
@@ -257,6 +256,8 @@ class WeightDtype(Enum):
             return WeightDtype.int3        
         if weight_dtype in ["int4"]:
             return WeightDtype.int4
+        if weight_dtype in ["int8"]:
+            return WeightDtype.int8        
         if weight_dtype in ["nf4"]:
             return WeightDtype.nf4
         if weight_dtype in ["fp4"]:
@@ -264,11 +265,11 @@ class WeightDtype(Enum):
         if weight_dtype in ["All"]:
             return WeightDtype.all
         if weight_dtype in ["float16"]:
-            return WeightDtype.fp16
+            return WeightDtype.f16
         if weight_dtype in ["bfloat16"]:
             return WeightDtype.bf16
         if weight_dtype in ["float32"]:
-            return WeightDtype.fp32
+            return WeightDtype.f32        
         return WeightDtype.Unknown
 
 class ComputeDtype(Enum):