Skip to content

Commit f1c317c

Browse files
committed
[SPARKNLP-1091] AutoGGUFEmbeddings Set Defaults
1 parent b59a339 commit f1c317c

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/main/scala/com/johnsnowlabs/nlp/embeddings/AutoGGUFEmbeddings.scala

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,12 @@ class AutoGGUFEmbeddings(override val uid: String)
135135

136136
private[johnsnowlabs] def setEngine(engineName: String): this.type = set(engine, engineName)
137137

138-
setDefault(engine -> LlamaCPP.name, embedding -> true, poolingType -> "MEAN", nCtx -> 512)
138+
setDefault(
139+
engine -> LlamaCPP.name,
140+
embedding -> true,
141+
poolingType -> "MEAN",
142+
nCtx -> 4096,
143+
nBatch -> 512)
139144

140145
override def onWrite(path: String, spark: SparkSession): Unit = {
141146
super.onWrite(path, spark)

0 commit comments

Comments
 (0)