Fix mxfp4 dequantization (#41292)

Cyrilvallez · ArthurZucker · commit aca2380b1b50 · 2025-10-03T12:04:26.000+02:00
fix
diff --git a/src/transformers/integrations/mxfp4.py b/src/transformers/integrations/mxfp4.py
@@ -343,7 +343,6 @@ def dequantize(module, param_name, param_value, target_device, dq_param_name, **
                     to_contiguous,
                     rank,
                     device_mesh,
-                    set_param=False,
                 )
             blocks_attr = f"{proj}_blocks"
             scales_attr = f"{proj}_scales"

Original file line number	Diff line number	Diff line change
`@@ -343,7 +343,6 @@ def dequantize(module, param_name, param_value, target_device, dq_param_name, **`
`343`	`343`	`to_contiguous,`
`344`	`344`	`rank,`
`345`	`345`	`device_mesh,`
`346`		`- set_param=False,`
`347`	`346`	`)`
`348`	`347`	`blocks_attr = f"{proj}_blocks"`
`349`	`348`	`scales_attr = f"{proj}_scales"`