From 286746d7c5e4765730b35ea57cc68d4ed862bef2 Mon Sep 17 00:00:00 2001 From: Maziyar Panahi Date: Wed, 20 Mar 2024 15:42:00 +0100 Subject: [PATCH] Fix XLMRoBertaForXX : revert the change in tfFile naming (#14204) * revert the change in tfFile naming - revert the change introduced in https://github.com/JohnSnowLabs/spark-nlp/pull/14130 * renaming tf file name in reading * Revert "renaming tf file name in reading" This reverts commit 7e5d33e7a325a995f9180ac078b4c58fd27bda8b. --------- Co-authored-by: ahmedlone127 --- .../classifier/dl/XlmRoBertaForQuestionAnswering.scala | 2 +- .../classifier/dl/XlmRoBertaForSequenceClassification.scala | 2 +- .../classifier/dl/XlmRoBertaForTokenClassification.scala | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala index 3cb512d56195b9..b885f640d94145 100644 --- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala +++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForQuestionAnswering.scala @@ -306,7 +306,7 @@ trait ReadXlmRoBertaForQuestionAnsweringDLModel with ReadSentencePieceModel { this: ParamsAndFeaturesReadable[XlmRoBertaForQuestionAnswering] => - override val tfFile: String = "xlm_roberta_classification_tf" + override val tfFile: String = "xlm_roberta_classification_tensorflow" override val onnxFile: String = "xlm_roberta_classification_onnx" override val sppFile: String = "xlmroberta_spp" diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala index 8f9f12c5a2af8f..366b0ce0fa8ad6 100644 --- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala +++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForSequenceClassification.scala @@ -357,7 +357,7 @@ trait ReadXlmRoBertaForSequenceDLModel with ReadSentencePieceModel { this: ParamsAndFeaturesReadable[XlmRoBertaForSequenceClassification] => - override val tfFile: String = "xlm_roberta_classification_tf" + override val tfFile: String = "xlm_roberta_classification_tensorflow" override val onnxFile: String = "xlm_roberta_classification_onnx" override val sppFile: String = "xlmroberta_spp" diff --git a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala index 2f5328583a2712..f9f933c1f8d018 100644 --- a/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala +++ b/src/main/scala/com/johnsnowlabs/nlp/annotators/classifier/dl/XlmRoBertaForTokenClassification.scala @@ -328,7 +328,7 @@ trait ReadXlmRoBertaForTokenDLModel with ReadSentencePieceModel { this: ParamsAndFeaturesReadable[XlmRoBertaForTokenClassification] => - override val tfFile: String = "xlm_roberta_classification_tf" + override val tfFile: String = "xlm_roberta_classification_tensorflow" override val onnxFile: String = "xlm_roberta_classification_onnx" override val sppFile: String = "xlmroberta_spp"