Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
danielz02
commited on
Fix capitalization for model precision
Browse files- 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 ["
|
118 |
return Precision.qt_gptq_3bit
|
119 |
-
if precision in ["
|
120 |
return Precision.qt_gptq_4bit
|
121 |
-
if precision in ["
|
122 |
return Precision.qt_gptq_8bit
|
123 |
-
if precision in ["
|
124 |
return Precision.qt_awq_3bit
|
125 |
-
if precision in ["
|
126 |
return Precision.qt_awq_4bit
|
127 |
-
if precision in ["
|
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 |
|