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__(
232232 "Currently only single quantization group is supported, please raise an issue with model details for support!"
233233 )
234234
235- if quantization_status != " frozen" :
235+ if quantization_status not in { "compressed" , " frozen"} :
236236 raise NotImplementedError (f"expected quantization_status=`frozen`, got { quantization_status } " )
237237
238238 if kv_cache_scheme :
239239 raise NotImplementedError (f"Expected kv_cache_scheme=None, got { kv_cache_scheme } " )
240240
241- if format not in [ "naive-quantized" , "float-quantized" ] :
241+ if format not in { "naive-quantized" , "float-quantized" } :
242242 raise NotImplementedError (
243243 f"Expected quantization format in ['naive_quantized', 'float-quantized'] got { format } "
244244 )
You can’t perform that action at this time.
0 commit comments