Skip to content

Commit 1e1481d

Browse files
ahmedlone127DevinTDHa
authored andcommitted
updating default model name
1 parent 1408aa3 commit 1e1481d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

python/sparknlp/annotator/cv/florence2_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def loadSavedModel(folder, spark_session, use_openvino=False):
174174
return Florence2Transformer(java_model=jModel)
175175

176176
@staticmethod
177-
def pretrained(name="florence2_base_ft_int4", lang="en", remote_loc=None,engine ="onnx"):
177+
def pretrained(name="florence_2_base_ft_int4", lang="en", remote_loc=None,engine ="onnx"):
178178
"""Downloads and loads a pretrained model."""
179179
from sparknlp.pretrained import ResourceDownloader
180180
return ResourceDownloader.downloadModel(Florence2Transformer, name, lang, remote_loc,engine)

src/main/scala/com/johnsnowlabs/nlp/annotators/cv/Florence2Transformer.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ class Florence2Transformer(override val uid: String)
330330
trait ReadablePretrainedFlorence2TransformerModel
331331
extends ParamsAndFeaturesReadable[Florence2Transformer]
332332
with HasPretrained[Florence2Transformer] {
333-
override val defaultModelName: Some[String] = Some("florence2_base_ft_int4")
333+
override val defaultModelName: Some[String] = Some("florence_2_base_ft_int4")
334334

335335
/** Java compliant-overrides */
336336
override def pretrained(): Florence2Transformer = super.pretrained()

0 commit comments

Comments
 (0)