Skip to content

Commit 0f69807

Browse files
committed
smol fixes.
1 parent 180c121 commit 0f69807

File tree

1 file changed

+5
-4
lines changed
  • src/diffusers/quantizers/bitsandbytes

1 file changed

+5
-4
lines changed

src/diffusers/quantizers/bitsandbytes/utils.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)