Skip to content

Commit

Permalink
Fix a bug that prevents specifying include patterns for quantifying
Browse files Browse the repository at this point in the history
  • Loading branch information
kaibioinfo authored and dacorvo committed Aug 14, 2024
1 parent 601dc19 commit d7cac04
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/quanto/quantize.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def quantize(
any patterns from the denylist.
"""
if include is not None:
include = [include] if isinstance(include, str) else exclude
include = [include] if isinstance(include, str) else include
if exclude is not None:
exclude = [exclude] if isinstance(exclude, str) else exclude
for name, m in model.named_modules():
Expand Down

0 comments on commit d7cac04

Please sign in to comment.