From 7e5d33e7a325a995f9180ac078b4c58fd27bda8b Mon Sep 17 00:00:00 2001 From: ahmedlone127 Date: Sun, 17 Mar 2024 03:46:29 +0500 Subject: [PATCH] renaming tf file name in reading --- .../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 b885f640d94145..bd508a4d7b1a9f 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 @@ -319,7 +319,7 @@ trait ReadXlmRoBertaForQuestionAnsweringDLModel instance.getEngine match { case TensorFlow.name => val tfWrapper = - readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false) + readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false) instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp) case ONNX.name => val onnxWrapper = 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 366b0ce0fa8ad6..782cbd01ed5be3 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 @@ -370,7 +370,7 @@ trait ReadXlmRoBertaForSequenceDLModel instance.getEngine match { case TensorFlow.name => val tfWrapper = - readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false) + readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false) instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp) case ONNX.name => val onnxWrapper = 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 f9f933c1f8d018..2f6cbed1db5f5a 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 @@ -342,7 +342,7 @@ trait ReadXlmRoBertaForTokenDLModel instance.getEngine match { case TensorFlow.name => val tfWrapper = - readTensorflowModel(path, spark, "xlm_roberta_classification_tf", initAllTables = false) + readTensorflowModel(path, spark, "xlm_roberta_classification_tensorflow", initAllTables = false) instance.setModelIfNotSet(spark, Some(tfWrapper), None, spp) case ONNX.name => val onnxWrapper =