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 |
|