Skip to content

Commit 5417d91

Browse files
authored
updating python and scala model names (#14488)
1 parent c31306f commit 5417d91

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

python/sparknlp/annotator/seq2seq/llama3_transformer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ class LLAMA3Transformer(AnnotatorModel, HasBatchedAnnotate, HasEngine):
3838
... .setOutputCol("generation")
3939
4040
41-
The default model is ``"llama3-7b"``, if no name is provided. For available
41+
The default model is ``"llama_3_7b_chat_hf_int4"``, if no name is provided. For available
4242
pretrained models please see the `Models Hub
4343
<https://sparknlp.org/models?q=llama3>`__.
4444
@@ -108,7 +108,7 @@ class LLAMA3Transformer(AnnotatorModel, HasBatchedAnnotate, HasEngine):
108108
>>> documentAssembler = DocumentAssembler() \\
109109
... .setInputCol("text") \\
110110
... .setOutputCol("documents")
111-
>>> llama3 = LLAMA3Transformer.pretrained("llama_3_7b_chat_hf_int8") \\
111+
>>> llama3 = LLAMA3Transformer.pretrained("llama_3_7b_chat_hf_int4") \\
112112
... .setInputCols(["documents"]) \\
113113
... .setMaxOutputLength(60) \\
114114
... .setOutputCol("generation")
@@ -365,7 +365,7 @@ def pretrained(name="llama_3_7b_chat_hf_int4", lang="en", remote_loc=None):
365365
Parameters
366366
----------
367367
name : str, optional
368-
Name of the pretrained model, by default "llama_2_7b_chat_hf_int4"
368+
Name of the pretrained model, by default "llama_3_7b_chat_hf_int4"
369369
lang : str, optional
370370
Language of the pretrained model, by default "en"
371371
remote_loc : str, optional

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ import org.json4s.jackson.JsonMethods._
6565
* .setInputCols("document")
6666
* .setOutputCol("generation")
6767
* }}}
68-
* The default model is `"llama_3_7b_chat_hf_int8"`, if no name is provided. For available
68+
* The default model is `"llama_3_7b_chat_hf_int4"`, if no name is provided. For available
6969
* pretrained models please see the [[https://sparknlp.org/models?q=llama3 Models Hub]].
7070
*
7171
* For extended examples of usage, see
@@ -101,7 +101,7 @@ import org.json4s.jackson.JsonMethods._
101101
* .setInputCol("text")
102102
* .setOutputCol("documents")
103103
*
104-
* val llama3 = LLAMA3Transformer.pretrained("llama_3_7b_chat_hf_int8")
104+
* val llama3 = LLAMA3Transformer.pretrained("llama_3_7b_chat_hf_int4")
105105
* .setInputCols(Array("documents"))
106106
* .setMinOutputLength(15)
107107
* .setMaxOutputLength(60)
@@ -359,7 +359,7 @@ class LLAMA3Transformer(override val uid: String)
359359
trait ReadablePretrainedLLAMA3TransformerModel
360360
extends ParamsAndFeaturesReadable[LLAMA3Transformer]
361361
with HasPretrained[LLAMA3Transformer] {
362-
override val defaultModelName: Some[String] = Some("llama3")
362+
override val defaultModelName: Some[String] = Some("llama_3_7b_chat_hf_int4")
363363

364364
/** Java compliant-overrides */
365365
override def pretrained(): LLAMA3Transformer = super.pretrained()

0 commit comments

Comments
 (0)