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 037b7d5 commit 10b773aCopy full SHA for 10b773a
src/main/scala/com/johnsnowlabs/nlp/annotators/seq2seq/AutoGGUFModel.scala
@@ -156,6 +156,8 @@ class AutoGGUFModel(override val uid: String)
156
157
private[johnsnowlabs] def setEngine(engineName: String): this.type = set(engine, engineName)
158
159
+ setDefault(engine -> LlamaCPP.name)
160
+
161
override def onWrite(path: String, spark: SparkSession): Unit = {
162
super.onWrite(path, spark)
163
getModelIfNotSet.saveToFile(path)
0 commit comments