@@ -58,8 +58,8 @@ class BitWidthConfig:
5858 manual_weights_bit_width_selection_list : List [ManualWeightsBitWidthSelection ] = field (default_factory = list )
5959
6060 def set_manual_activation_bit_width (self ,
61- filters : Union [List [BaseNodeMatcher ], BaseNodeMatcher ],
62- bit_widths : Union [List [int ], int ]):
61+ filters : Union [List [BaseNodeMatcher ], BaseNodeMatcher ],
62+ bit_widths : Union [List [int ], int ]):
6363 """
6464 Add a manual bit-width selection for activation to the configuration.
6565
@@ -77,8 +77,7 @@ def set_manual_activation_bit_width(self,
7777 def set_manual_weights_bit_width (self ,
7878 filters : Union [List [BaseNodeMatcher ], BaseNodeMatcher ],
7979 bit_widths : Union [List [int ], int ],
80- attrs : Union [List [WeightAttrT ], WeightAttrT ]
81- ):
80+ attrs : Union [List [WeightAttrT ], WeightAttrT ]):
8281 """
8382 Add a manual bit-width selection for weights to the configuration.
8483
@@ -193,7 +192,7 @@ def _construct_node_to_new_bit_mapping(graph, manual_bit_width_selection_list) -
193192 unit_nodes_to_change_bit_width .update ({n : manual_bit_width_selection .bit_width })
194193 elif type (manual_bit_width_selection ) is ManualWeightsBitWidthSelection :
195194 if len (n .get_node_weights_attributes ()) == 0 :
196- Logger .critical (f'The requested attribute to change the bit width for { n } is not existing .' )
195+ Logger .critical (f'The requested attribute to change the bit width for { n } does not exist .' )
197196
198197 if n in unit_nodes_to_change_bit_width :
199198 if unit_nodes_to_change_bit_width [n ][1 ] == manual_bit_width_selection .attr :
0 commit comments