danielz02 commited on
Commit
4d75aa7
1 Parent(s): 300e6b5

Fix capitalization for model precision

Browse files
Files changed (1) hide show
  1. src/display/utils.py +6 -6
src/display/utils.py CHANGED
@@ -114,17 +114,17 @@ class Precision(Enum):
114
  return Precision.float16
115
  if precision in ["torch.bfloat16", "bfloat16"]:
116
  return Precision.bfloat16
117
- if precision in ["gptq-3bit"]:
118
  return Precision.qt_gptq_3bit
119
- if precision in ["gptq-4bit"]:
120
  return Precision.qt_gptq_4bit
121
- if precision in ["gptq-8bit"]:
122
  return Precision.qt_gptq_8bit
123
- if precision in ["awq-3bit"]:
124
  return Precision.qt_awq_3bit
125
- if precision in ["awq-4bit"]:
126
  return Precision.qt_awq_4bit
127
- if precision in ["awq-8bit"]:
128
  return Precision.qt_awq_8bit
129
  return Precision.Unknown
130
 
 
114
  return Precision.float16
115
  if precision in ["torch.bfloat16", "bfloat16"]:
116
  return Precision.bfloat16
117
+ if precision in ["GPTQ-3bit"]:
118
  return Precision.qt_gptq_3bit
119
+ if precision in ["GPTQ-4bit"]:
120
  return Precision.qt_gptq_4bit
121
+ if precision in ["GPTQ-8bit"]:
122
  return Precision.qt_gptq_8bit
123
+ if precision in ["AWQ-3bit"]:
124
  return Precision.qt_awq_3bit
125
+ if precision in ["AWQ-4bit"]:
126
  return Precision.qt_awq_4bit
127
+ if precision in ["AWQ-8bit"]:
128
  return Precision.qt_awq_8bit
129
  return Precision.Unknown
130