diff --git a/neural_compressor/adaptor/torch_utils/util.py b/neural_compressor/adaptor/torch_utils/util.py index 59f1be62b65..973e3a80787 100644 --- a/neural_compressor/adaptor/torch_utils/util.py +++ b/neural_compressor/adaptor/torch_utils/util.py @@ -250,7 +250,7 @@ def check_cfg_and_qconfig(tune_cfg, cfgs, op_infos_from_cfgs, output_tensor_ids_ if pre_op_output['id'] == input_tensor_id: pre_op_output_infos[index]['inf_dtype'] = input_tensor_dtype else: - print('Do not find the input id', input_tensor_id) + pass pre_op_infos['output_tensor_infos'] = pre_op_output_infos cfgs[pre_op_module][pre_op_state][pre_op_index] = pre_op_infos else: @@ -958,4 +958,4 @@ def collect_weight_info(q_config): 'group_size': config['weight']['group_size'], 'scheme': config['weight']['scheme'], } - return weight_info \ No newline at end of file + return weight_info