Skip to content

Commit

Permalink
Update model_utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zhijianma authored Nov 28, 2023
1 parent ccc26ad commit d0f564f
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions data_juicer/utils/model_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,6 +185,7 @@ def prepare_huggingface_clip(clip_name):

return (model, processor)


def prepare_huggingface_blip(blip_name):
"""
Prepare and load a blip and processor from HuggingFace.
Expand All @@ -200,6 +201,7 @@ def prepare_huggingface_blip(blip_name):

return (model, processor)


def prepare_diversity_model(model_name, lang):
"""
Prepare diversity model for specific language.
Expand Down

0 comments on commit d0f564f

Please sign in to comment.