Skip to content

Commit 427de37

Browse files
authored
[SPARKNLP-1300] changing token sequence in warmup test (#14677)
* changing token sequence in warmup test * updating warmup token sequence
1 parent 67a4810 commit 427de37

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/main/scala/com/johnsnowlabs/ml/ai/RoBerta.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ private[johnsnowlabs] class RoBerta(
7070

7171
private def sessionWarmup(): Unit = {
7272
val dummyInput =
73-
Array(0, 7939, 18, 3279, 658, 5, 19374, 13, 5, 78, 42752, 4, 2)
73+
Array(0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2)
7474
if (modelArch == ModelArch.wordEmbeddings) {
7575
tag(Seq(dummyInput))
7676
} else if (modelArch == ModelArch.sentenceEmbeddings) {

0 commit comments

Comments
 (0)