From 0842c7509714135e58c9515a8c2b9a4d0a096bd6 Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Tue, 3 Feb 2026 15:39:03 +0000 Subject: [PATCH] tmp fix Signed-off-by: yiliu30 --- auto_round/compressors/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/auto_round/compressors/base.py b/auto_round/compressors/base.py index 8e6d1fbae..0f0414bfc 100644 --- a/auto_round/compressors/base.py +++ b/auto_round/compressors/base.py @@ -1451,6 +1451,7 @@ def _quantize_via_rtn_blockwise(self, all_to_quantized_module_names: list[str]) materialize_model_(block) block.to("cpu") convert_module_to_hp_if_necessary(block, dtype=self.amp_dtype, device=self.device) + self._register_act_max_hook(block) if is_auto_device_mapping(self.device_map) and len(self.device_list) > 1: set_auto_device_map_for_block_with_tuning(