Unverified Commit 34e0cc6d authored by mustapha ajeghrir's avatar mustapha ajeghrir Committed by GitHub
Browse files

Make `Camembert` TF version independent from `Roberta` (#19364)



* camembert tf version independent

* fixup

* fixup, all working

* remove comments

* Adding copied from roberta
Co-authored-by: default avatarMustapha AJEGHRIR <mustapha.ajeghrir@kleegroup.com>
parent 7418a48e
...@@ -2314,6 +2314,7 @@ else: ...@@ -2314,6 +2314,7 @@ else:
"TFCamembertForSequenceClassification", "TFCamembertForSequenceClassification",
"TFCamembertForTokenClassification", "TFCamembertForTokenClassification",
"TFCamembertModel", "TFCamembertModel",
"TFCamembertPreTrainedModel",
] ]
) )
_import_structure["models.clip"].extend( _import_structure["models.clip"].extend(
...@@ -4981,6 +4982,7 @@ if TYPE_CHECKING: ...@@ -4981,6 +4982,7 @@ if TYPE_CHECKING:
TFCamembertForSequenceClassification, TFCamembertForSequenceClassification,
TFCamembertForTokenClassification, TFCamembertForTokenClassification,
TFCamembertModel, TFCamembertModel,
TFCamembertPreTrainedModel,
) )
from .models.clip import ( from .models.clip import (
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST, TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST,
......
...@@ -81,6 +81,7 @@ else: ...@@ -81,6 +81,7 @@ else:
"TFCamembertForSequenceClassification", "TFCamembertForSequenceClassification",
"TFCamembertForTokenClassification", "TFCamembertForTokenClassification",
"TFCamembertModel", "TFCamembertModel",
"TFCamembertPreTrainedModel",
] ]
...@@ -136,6 +137,7 @@ if TYPE_CHECKING: ...@@ -136,6 +137,7 @@ if TYPE_CHECKING:
TFCamembertForSequenceClassification, TFCamembertForSequenceClassification,
TFCamembertForTokenClassification, TFCamembertForTokenClassification,
TFCamembertModel, TFCamembertModel,
TFCamembertPreTrainedModel,
) )
else: else:
......
...@@ -637,6 +637,13 @@ class TFCamembertModel(metaclass=DummyObject): ...@@ -637,6 +637,13 @@ class TFCamembertModel(metaclass=DummyObject):
requires_backends(self, ["tf"]) requires_backends(self, ["tf"])
class TFCamembertPreTrainedModel(metaclass=DummyObject):
_backends = ["tf"]
def __init__(self, *args, **kwargs):
requires_backends(self, ["tf"])
TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST = None TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LIST = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment