Skip to content

Commit 765781a

Browse files
committed
add resource downloader
Signed-off-by: Prabod Rathnayaka <prabod@rathnayaka.me>
1 parent 6994b36 commit 765781a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -710,7 +710,8 @@ object PythonResourceDownloader {
710710
"PaliGemmaForMultiModal" -> PaliGemmaForMultiModal,
711711
"Gemma3ForMultiModal" -> Gemma3ForMultiModal,
712712
"InternVLForMultiModal" -> InternVLForMultiModal,
713-
"Florence2Transformer" -> Florence2Transformer)
713+
"Florence2Transformer" -> Florence2Transformer,
714+
"E5VEmbeddings" -> E5VEmbeddings)
714715

715716
// List pairs of types such as the one with key type can load a pretrained model from the value type
716717
val typeMapper: Map[String, String] = Map("ZeroShotNerModel" -> "RoBertaForQuestionAnswering")

0 commit comments

Comments
 (0)