Skip to content

Commit

Permalink
fix: remove extra prepare model when enable tokenization flag in some…
Browse files Browse the repository at this point in the history
… ops (#99)
  • Loading branch information
zhijianma authored Nov 24, 2023
1 parent 155f9f6 commit 754720a
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions data_juicer/utils/model_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,6 +269,4 @@ def get_model(model_key, lang='en', model_type='sentencepiece'):
:param model_key: name of the model or tokenzier
"""
if model_key not in MODEL_ZOO:
prepare_model(lang=lang, model_type=model_type, model_key=model_key)
return MODEL_ZOO.get(model_key, None)

0 comments on commit 754720a

Please sign in to comment.