File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
src/diffusers/quantizers/bitsandbytes Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -267,9 +267,10 @@ def _dequantize_and_replace(
267267 if len (list (module .children ())) > 0 :
268268 _ , has_been_replaced = _dequantize_and_replace (
269269 module ,
270- modules_to_not_convert ,
271- current_key_name ,
272- quantization_config ,
270+ dtype = dtype ,
271+ modules_to_not_convert = modules_to_not_convert ,
272+ current_key_name = current_key_name ,
273+ quantization_config = quantization_config ,
273274 has_been_replaced = has_been_replaced ,
274275 )
275276 # Remove the last key for recursion
@@ -284,7 +285,7 @@ def dequantize_and_replace(
284285):
285286 model , has_been_replaced = _dequantize_and_replace (
286287 model ,
287- model .dtype ,
288+ dtype = model .dtype ,
288289 modules_to_not_convert = modules_to_not_convert ,
289290 quantization_config = quantization_config ,
290291 )
You can’t perform that action at this time.
0 commit comments