[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
5ef000f912
commit
9c5b35c710
2 changed files with 4 additions and 1 deletions
|
|
@ -86,7 +86,9 @@ def main(argv = None) -> int:
|
|||
quantize_(
|
||||
transformer,
|
||||
_make_quant_config(scheme),
|
||||
filter_fn = make_filter_fn(args.min_features, exclude_name_tokens = exclude_tokens_for_scheme(scheme)),
|
||||
filter_fn = make_filter_fn(
|
||||
args.min_features, exclude_name_tokens = exclude_tokens_for_scheme(scheme)
|
||||
),
|
||||
)
|
||||
|
||||
# Move the state dict to CPU for a portable, GPU-free artifact.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue