dummy_flax_objects.py 896 Bytes
Newer Older
1
2
3
4
# This file is autogenerated by the command `make fix-copies`, do not edit.
from ..file_utils import requires_flax


Sylvain Gugger's avatar
Sylvain Gugger committed
5
6
7
8
9
10
11
12
13
14
15
16
FLAX_MODEL_MAPPING = None


class FlaxAutoModel:
    def __init__(self, *args, **kwargs):
        requires_flax(self)

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_flax(self)


17
18
19
20
21
22
23
24
25
class FlaxBertForMaskedLM:
    def __init__(self, *args, **kwargs):
        requires_flax(self)

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_flax(self)


26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
class FlaxBertModel:
    def __init__(self, *args, **kwargs):
        requires_flax(self)

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_flax(self)


class FlaxRobertaModel:
    def __init__(self, *args, **kwargs):
        requires_flax(self)

    @classmethod
    def from_pretrained(self, *args, **kwargs):
        requires_flax(self)