Skip to content

Commit 10b773a

Browse files
committed
add a default engine to the AutoGGUFModel
1 parent 037b7d5 commit 10b773a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/main/scala/com/johnsnowlabs/nlp/annotators/seq2seq/AutoGGUFModel.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,8 @@ class AutoGGUFModel(override val uid: String)
156156

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

159+
setDefault(engine -> LlamaCPP.name)
160+
159161
override def onWrite(path: String, spark: SparkSession): Unit = {
160162
super.onWrite(path, spark)
161163
getModelIfNotSet.saveToFile(path)

0 commit comments

Comments
 (0)