Skip to content

Commit

Permalink
remove invalid print (#1057)
Browse files Browse the repository at this point in the history
  • Loading branch information
changwangss authored Jul 4, 2023
1 parent 94df997 commit 6c30464
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions neural_compressor/adaptor/torch_utils/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -958,4 +958,4 @@ def collect_weight_info(q_config):
'group_size': config['weight']['group_size'],
'scheme': config['weight']['scheme'],
}
return weight_info
return weight_info

0 comments on commit 6c30464

Please sign in to comment.