Skip to content

Commit b54f790

Browse files
authored
replace source_keys and target_keys (#42471)
fix
1 parent 6db4332 commit b54f790

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

src/transformers/quantizers/quantizer_bnb_4bit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -306,7 +306,7 @@ def get_weight_conversions(self):
306306
if self.pre_quantized:
307307
return [
308308
WeightConverter(
309-
source_keys=[
309+
source_patterns=[
310310
"weight.nested_absmax",
311311
"weight.nested_quant_map",
312312
"weight.quant_map",
@@ -315,7 +315,7 @@ def get_weight_conversions(self):
315315
"weight.quant_state.bitsandbytes__fp4",
316316
"weight",
317317
],
318-
target_keys="weight",
318+
target_patterns="weight",
319319
operations=[Bnb4bitDeserialize(self)],
320320
)
321321
]

src/transformers/quantizers/quantizer_bnb_8bit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,8 +254,8 @@ def get_weight_conversions(self):
254254
if self.pre_quantized:
255255
return [
256256
WeightConverter(
257-
source_keys=["SCB", "weight_format", "weight"],
258-
target_keys="weight",
257+
source_patterns=["SCB", "weight_format", "weight"],
258+
target_patterns="weight",
259259
operations=[Bnb8bitDeserialize(self)],
260260
)
261261
]

src/transformers/quantizers/quantizer_mxfp4.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -443,16 +443,16 @@ def get_weight_conversions(self):
443443
if self.quantization_config.dequantize:
444444
return [
445445
WeightConverter(
446-
source_keys=["_blocks", "_scales"],
447-
target_keys="",
446+
source_patterns=["_blocks", "_scales"],
447+
target_patterns="",
448448
operations=[Mxfp4Dequantize(self)],
449449
)
450450
]
451451
else:
452452
return [
453453
WeightConverter(
454-
source_keys=["_blocks", "_scales"],
455-
target_keys="",
454+
source_patterns=["_blocks", "_scales"],
455+
target_patterns="",
456456
operations=[Mxfp4Deserialize(self)],
457457
)
458458
]

0 commit comments

Comments
 (0)