Update src/display/utils.py
Browse files- src/display/utils.py +4 -0
src/display/utils.py
CHANGED
|
@@ -210,6 +210,8 @@ class QuantType(Enum):
|
|
| 210 |
bnb = ModelDetails(name="bitsandbytes", symbol="π΅")
|
| 211 |
autoround = ModelDetails(name="AutoRound", symbol="π")
|
| 212 |
Unknown = ModelDetails(name="?", symbol="?")
|
|
|
|
|
|
|
| 213 |
|
| 214 |
def to_str(self, separator=" "):
|
| 215 |
return f"{self.value.symbol}{separator}{self.value.name}"
|
|
@@ -227,6 +229,8 @@ class QuantType(Enum):
|
|
| 227 |
return QuantType.bnb
|
| 228 |
if quant_dtype in ["AutoRound"]:
|
| 229 |
return QuantType.autoround
|
|
|
|
|
|
|
| 230 |
return QuantType.Unknown
|
| 231 |
|
| 232 |
|
|
|
|
| 210 |
bnb = ModelDetails(name="bitsandbytes", symbol="π΅")
|
| 211 |
autoround = ModelDetails(name="AutoRound", symbol="π")
|
| 212 |
Unknown = ModelDetails(name="?", symbol="?")
|
| 213 |
+
QuantType_None = ModelDetails(name="None", symbol="β")
|
| 214 |
+
|
| 215 |
|
| 216 |
def to_str(self, separator=" "):
|
| 217 |
return f"{self.value.symbol}{separator}{self.value.name}"
|
|
|
|
| 229 |
return QuantType.bnb
|
| 230 |
if quant_dtype in ["AutoRound"]:
|
| 231 |
return QuantType.autoround
|
| 232 |
+
if quant_dtype in ["None"]:
|
| 233 |
+
return QuantType.QuantType_None
|
| 234 |
return QuantType.Unknown
|
| 235 |
|
| 236 |
|