Skip to content

Commit 2a3c40c

Browse files
committed
Fixed comments on pull requests.
1 parent 8ee3002 commit 2a3c40c

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

model_compression_toolkit/core/common/quantization/bit_width_config.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)