"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "4c62c6021aa3341eff4c0bb87fea72fea22eb701"
Unverified Commit 48395d6b authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix init for MT5 (#8591)

parent a6cf9ca0
...@@ -145,6 +145,7 @@ from .models.marian import MarianConfig ...@@ -145,6 +145,7 @@ from .models.marian import MarianConfig
from .models.mbart import MBartConfig from .models.mbart import MBartConfig
from .models.mmbt import MMBTConfig from .models.mmbt import MMBTConfig
from .models.mobilebert import MOBILEBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, MobileBertConfig, MobileBertTokenizer from .models.mobilebert import MOBILEBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, MobileBertConfig, MobileBertTokenizer
from .models.mt5 import MT5Config
from .models.openai import OPENAI_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP, OpenAIGPTConfig, OpenAIGPTTokenizer from .models.openai import OPENAI_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP, OpenAIGPTConfig, OpenAIGPTTokenizer
from .models.pegasus import PegasusConfig from .models.pegasus import PegasusConfig
from .models.phobert import PhobertTokenizer from .models.phobert import PhobertTokenizer
...@@ -498,7 +499,7 @@ if is_torch_available(): ...@@ -498,7 +499,7 @@ if is_torch_available():
MobileBertPreTrainedModel, MobileBertPreTrainedModel,
load_tf_weights_in_mobilebert, load_tf_weights_in_mobilebert,
) )
from .models.mt5 import MT5Config, MT5ForConditionalGeneration, MT5Model from .models.mt5 import MT5ForConditionalGeneration, MT5Model
from .models.openai import ( from .models.openai import (
OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST, OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST,
OpenAIGPTDoubleHeadsModel, OpenAIGPTDoubleHeadsModel,
......
...@@ -1361,11 +1361,6 @@ def load_tf_weights_in_mobilebert(*args, **kwargs): ...@@ -1361,11 +1361,6 @@ def load_tf_weights_in_mobilebert(*args, **kwargs):
requires_pytorch(load_tf_weights_in_mobilebert) requires_pytorch(load_tf_weights_in_mobilebert)
class MT5Config:
def __init__(self, *args, **kwargs):
requires_pytorch(self)
class MT5ForConditionalGeneration: class MT5ForConditionalGeneration:
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
requires_pytorch(self) requires_pytorch(self)
......
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