From d0f564f01ef640e517fe11babe9b45956802a307 Mon Sep 17 00:00:00 2001 From: zhijianma Date: Tue, 28 Nov 2023 08:57:32 +0800 Subject: [PATCH] Update model_utils.py --- data_juicer/utils/model_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/data_juicer/utils/model_utils.py b/data_juicer/utils/model_utils.py index 5ab381771..e4038daac 100644 --- a/data_juicer/utils/model_utils.py +++ b/data_juicer/utils/model_utils.py @@ -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. @@ -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.