Unverified Commit 60db81ff authored by mustapha ajeghrir's avatar mustapha ajeghrir Committed by GitHub
Browse files

Making camembert independent from roberta, clean (#19337)


Co-authored-by: default avatarMustapha AJEGHRIR <mustapha.ajeghrir@kleegroup.com>
parent c54bb1ad
...@@ -1024,6 +1024,7 @@ else: ...@@ -1024,6 +1024,7 @@ else:
"CamembertForSequenceClassification", "CamembertForSequenceClassification",
"CamembertForTokenClassification", "CamembertForTokenClassification",
"CamembertModel", "CamembertModel",
"CamembertPreTrainedModel",
] ]
) )
_import_structure["models.canine"].extend( _import_structure["models.canine"].extend(
...@@ -3873,6 +3874,7 @@ if TYPE_CHECKING: ...@@ -3873,6 +3874,7 @@ if TYPE_CHECKING:
CamembertForSequenceClassification, CamembertForSequenceClassification,
CamembertForTokenClassification, CamembertForTokenClassification,
CamembertModel, CamembertModel,
CamembertPreTrainedModel,
) )
from .models.canine import ( from .models.canine import (
CANINE_PRETRAINED_MODEL_ARCHIVE_LIST, CANINE_PRETRAINED_MODEL_ARCHIVE_LIST,
......
...@@ -63,6 +63,7 @@ else: ...@@ -63,6 +63,7 @@ else:
"CamembertForSequenceClassification", "CamembertForSequenceClassification",
"CamembertForTokenClassification", "CamembertForTokenClassification",
"CamembertModel", "CamembertModel",
"CamembertPreTrainedModel",
] ]
try: try:
...@@ -117,6 +118,7 @@ if TYPE_CHECKING: ...@@ -117,6 +118,7 @@ if TYPE_CHECKING:
CamembertForSequenceClassification, CamembertForSequenceClassification,
CamembertForTokenClassification, CamembertForTokenClassification,
CamembertModel, CamembertModel,
CamembertPreTrainedModel,
) )
try: try:
......
...@@ -1093,6 +1093,13 @@ class CamembertModel(metaclass=DummyObject): ...@@ -1093,6 +1093,13 @@ class CamembertModel(metaclass=DummyObject):
requires_backends(self, ["torch"]) requires_backends(self, ["torch"])
class CamembertPreTrainedModel(metaclass=DummyObject):
_backends = ["torch"]
def __init__(self, *args, **kwargs):
requires_backends(self, ["torch"])
CANINE_PRETRAINED_MODEL_ARCHIVE_LIST = None CANINE_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