Skip to content

Commit efc4b2b

Browse files
irenabirenab
authored andcommitted
fix
1 parent a777853 commit efc4b2b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_compression_toolkit/core/common/graph/virtual_activation_weights_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def __init__(self,
169169
assert len(kernel_attrs) == 1 and kernel_attrs[0] is not None, 'Expected exactly one kernel attr.'
170170
kernel_attr = kernel_attrs[0]
171171
conf_attrs = [attr for attr in weights_node.weights if weights_node.is_configurable_weight(attr)]
172-
if len(conf_attrs) > 1 or len(conf_attrs) == 1 and conf_attrs[0] != kernel_attr: # pragma: no cover
172+
if len(conf_attrs) > 1 or len(conf_attrs) == 1 and not weights_node.is_configurable_weight(kernel_attr): # pragma: no cover
173173
raise NotImplementedError('Only kernel attr can be configurable.')
174174

175175
weights_candidates_quantization_cfg = weights_node.get_unique_weights_candidates(kernel_attr)

0 commit comments

Comments
 (0)