File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
QEfficient/transformers/quantizers Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -232,13 +232,13 @@ def __init__(
232
232
"Currently only single quantization group is supported, please raise an issue with model details for support!"
233
233
)
234
234
235
- if quantization_status != " frozen" :
235
+ if quantization_status not in { "compressed" , " frozen"} :
236
236
raise NotImplementedError (f"expected quantization_status=`frozen`, got { quantization_status } " )
237
237
238
238
if kv_cache_scheme :
239
239
raise NotImplementedError (f"Expected kv_cache_scheme=None, got { kv_cache_scheme } " )
240
240
241
- if format not in [ "naive-quantized" , "float-quantized" ] :
241
+ if format not in { "naive-quantized" , "float-quantized" } :
242
242
raise NotImplementedError (
243
243
f"Expected quantization format in ['naive_quantized', 'float-quantized'] got { format } "
244
244
)
You can’t perform that action at this time.
0 commit comments