Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
  • Loading branch information
stephantul committed Feb 4, 2025
1 parent 99fedd0 commit 984c38f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions model2vec/distill/distillation.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def distill_from_model(
:return: A StaticModel
"""
sif_coefficient = validate_parameters(tokenizer, vocabulary, apply_zipf, sif_coefficient, use_subword)
sif_coefficient = _validate_parameters(tokenizer, vocabulary, apply_zipf, sif_coefficient, use_subword)

device = select_optimal_device(device)
# Make a base list of tokens.
Expand Down Expand Up @@ -151,7 +151,7 @@ def distill_from_model(
)


def validate_parameters(
def _validate_parameters(
tokenizer: PreTrainedTokenizerFast,
vocabulary: list[str] | None,
apply_zipf: bool | None,
Expand Down

0 comments on commit 984c38f

Please sign in to comment.