Skip to content

Commit

Permalink
Fix XLMRoBertaForXX : revert the change in tfFile naming (#14204)
Browse files Browse the repository at this point in the history
* revert the change in tfFile naming

- revert the change introduced in #14130

* renaming tf file name in reading

* Revert "renaming tf file name in reading"

This reverts commit 7e5d33e.

---------

Co-authored-by: ahmedlone127 <ahmedlone127@gmail.com>
  • Loading branch information
maziyarpanahi and ahmedlone127 authored Mar 20, 2024
1 parent c587734 commit 286746d
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down

0 comments on commit 286746d

Please sign in to comment.