modify ComputeDtype params

src/display/utils.py

@@ -230,17 +230,17 @@ class QuantType(Enum):
         return QuantType.Unknown
 
 
+
 class WeightDtype(Enum):
+    all = ModelDetails("All")
     int2 = ModelDetails("int2")
-    int3 = ModelDetails("int3")
+    int3 = ModelDetails("int3")    
     int4 = ModelDetails("int4")
     nf4 = ModelDetails("nf4")
     fp4 = ModelDetails("fp4")
 
-
     Unknown = ModelDetails("?")
 
-    all = ModelDetails("All")
 
 
     def from_str(weight_dtype):
@@ -259,11 +259,13 @@ class WeightDtype(Enum):
         return WeightDtype.Unknown
 
 class ComputeDtype(Enum):
+    all = ModelDetails("All")
     fp16 = ModelDetails("float16")
     bf16 = ModelDetails("bfloat16")
     int8 = ModelDetails("int8")
     fp32 = ModelDetails("float32")
 
+
     Unknown = ModelDetails("?")
 
     def from_str(compute_dtype):
@@ -275,8 +277,11 @@ class ComputeDtype(Enum):
             return ComputeDtype.int8
         if compute_dtype in ["float32"]:
             return ComputeDtype.fp32
+        if compute_dtype in ["All"]:
+            return ComputeDtype.all        
         return ComputeDtype.Unknown
     
+    
 class GroupDtype(Enum):
     group_1 = ModelDetails("-1")
     group_1024 = ModelDetails("1024")