fix copies

huggingface · Jan 28, 2022 · d3bbe4f · d3bbe4f
1 parent 33c53e5
commit d3bbe4f
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 59 deletions.
diff --git a/src/transformers/utils/dummy_flax_objects.py b/src/transformers/utils/dummy_flax_objects.py
@@ -963,37 +963,22 @@ def __init__(self, *args, **kwargs):
         requires_backends(self, ["flax"])
 
 
-class FlaxXGLMForCausalLM:
+class FlaxXGLMForCausalLM(metaclass=DummyObject):
+    _backends = ["flax"]
+
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["flax"])
 
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["flax"])
-
-    def __call__(self, *args, **kwargs):
-        requires_backends(self, ["flax"])
 
+class FlaxXGLMModel(metaclass=DummyObject):
+    _backends = ["flax"]
 
-class FlaxXGLMModel:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["flax"])
 
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["flax"])
-
-    def __call__(self, *args, **kwargs):
-        requires_backends(self, ["flax"])
 
+class FlaxXGLMPreTrainedModel(metaclass=DummyObject):
+    _backends = ["flax"]
 
-class FlaxXGLMPreTrainedModel:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["flax"])
-
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["flax"])
-
-    def __call__(self, *args, **kwargs):
-        requires_backends(self, ["flax"])
diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py
@@ -3806,80 +3806,66 @@ def __init__(self, *args, **kwargs):
 WAVLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
 
 
-class XGLMForCausalLM:
+class WavLMForAudioFrameClassification(metaclass=DummyObject):
+    _backends = ["torch"]
+
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["torch"])
-
-    def forward(self, *args, **kwargs):
-        requires_backends(self, ["torch"])
 
+class WavLMForCTC(metaclass=DummyObject):
+    _backends = ["torch"]
 
-class XGLMModel:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["torch"])
-
-    def forward(self, *args, **kwargs):
-        requires_backends(self, ["torch"])
 
+class WavLMForSequenceClassification(metaclass=DummyObject):
+    _backends = ["torch"]
 
-class XGLMPreTrainedModel:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["torch"])
-
-    def forward(self, *args, **kwargs):
-        requires_backends(self, ["torch"])
-
 
-XLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
-
-class WavLMForAudioFrameClassification(metaclass=DummyObject):
+class WavLMForXVector(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
-class WavLMForCTC(metaclass=DummyObject):
+class WavLMModel(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
-class WavLMForSequenceClassification(metaclass=DummyObject):
+class WavLMPreTrainedModel(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
-class WavLMForXVector(metaclass=DummyObject):
+XGLM_PRETRAINED_MODEL_ARCHIVE_LIST = None
+
+
+class XGLMForCausalLM(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
-class WavLMModel(metaclass=DummyObject):
+class XGLMModel(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
-class WavLMPreTrainedModel(metaclass=DummyObject):
+class XGLMPreTrainedModel(metaclass=DummyObject):
     _backends = ["torch"]
 
     def __init__(self, *args, **kwargs):

diff --git a/src/transformers/utils/dummy_sentencepiece_objects.py b/src/transformers/utils/dummy_sentencepiece_objects.py
@@ -136,16 +136,16 @@ def __init__(self, *args, **kwargs):
         requires_backends(self, ["sentencepiece"])
 
 
-class XLMProphetNetTokenizer(metaclass=DummyObject):
+class XGLMTokenizer(metaclass=DummyObject):
     _backends = ["sentencepiece"]
 
-
-class XGLMTokenizer:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["sentencepiece"])
 
 
-class XLMProphetNetTokenizer:
+class XLMProphetNetTokenizer(metaclass=DummyObject):
+    _backends = ["sentencepiece"]
+
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["sentencepiece"])
 

diff --git a/src/transformers/utils/dummy_tokenizers_objects.py b/src/transformers/utils/dummy_tokenizers_objects.py
@@ -293,13 +293,13 @@ def __init__(self, *args, **kwargs):
 class T5TokenizerFast(metaclass=DummyObject):
     _backends = ["tokenizers"]
 
-
-class XGLMTokenizerFast:
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["tokenizers"])
 
 
-class XLMRobertaTokenizerFast:
+class XGLMTokenizerFast(metaclass=DummyObject):
+    _backends = ["tokenizers"]
+
     def __init__(self, *args, **kwargs):
         requires_backends(self, ["tokenizers"])