Skip to content

Commit 2d5bfa1

Browse files
AbdullahMubeenAnwarDevinTDHa
authored andcommitted
Specify model name in E5VEmbeddings test
default model no longer exists
1 parent 1e1481d commit 2d5bfa1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/test/scala/com/johnsnowlabs/nlp/embeddings/E5VEmbeddingsTestSpec.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ class E5VEmbeddingsTestSpec extends AnyFlatSpec {
6969
.setOutputCol("image_assembler")
7070

7171
val loadModel = E5VEmbeddings
72-
.pretrained()
72+
.pretrained("e5v_int4", "en")
7373
.setInputCols("image_assembler")
7474
.setOutputCol("e5v")
7575

0 commit comments

Comments
 (0)