wenjiao commited on
Commit
359bb98
β€’
1 Parent(s): d57a7e2

Update src/display/utils.py

Browse files
Files changed (1) hide show
  1. 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