We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 67a4810 commit 427de37Copy full SHA for 427de37
src/main/scala/com/johnsnowlabs/ml/ai/RoBerta.scala
@@ -70,7 +70,7 @@ private[johnsnowlabs] class RoBerta(
70
71
private def sessionWarmup(): Unit = {
72
val dummyInput =
73
- Array(0, 7939, 18, 3279, 658, 5, 19374, 13, 5, 78, 42752, 4, 2)
+ Array(0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2)
74
if (modelArch == ModelArch.wordEmbeddings) {
75
tag(Seq(dummyInput))
76
} else if (modelArch == ModelArch.sentenceEmbeddings) {
0 commit comments