Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
15e17c99
Unverified
Commit
15e17c99
authored
Jan 05, 2023
by
Sujay
Committed by
GitHub
Jan 04, 2023
Browse files
Remove T5 dependency from mT5 model (#20949)
make mt5 independent from t5
parent
9dcc881f
Changes
5
Expand all
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
1860 additions
and
26 deletions
+1860
-26
src/transformers/__init__.py
src/transformers/__init__.py
+4
-2
src/transformers/models/mt5/__init__.py
src/transformers/models/mt5/__init__.py
+8
-2
src/transformers/models/mt5/modeling_mt5.py
src/transformers/models/mt5/modeling_mt5.py
+1840
-22
src/transformers/utils/dummy_pt_objects.py
src/transformers/utils/dummy_pt_objects.py
+7
-0
utils/check_repo.py
utils/check_repo.py
+1
-0
No files found.
src/transformers/__init__.py
View file @
15e17c99
...
...
@@ -1806,7 +1806,9 @@ else:
"MPNetPreTrainedModel"
,
]
)
_import_structure
[
"models.mt5"
].
extend
([
"MT5EncoderModel"
,
"MT5ForConditionalGeneration"
,
"MT5Model"
])
_import_structure
[
"models.mt5"
].
extend
(
[
"MT5EncoderModel"
,
"MT5ForConditionalGeneration"
,
"MT5Model"
,
"MT5PreTrainedModel"
]
)
_import_structure
[
"models.mvp"
].
extend
(
[
"MVP_PRETRAINED_MODEL_ARCHIVE_LIST"
,
...
...
@@ -4922,7 +4924,7 @@ if TYPE_CHECKING:
MPNetModel
,
MPNetPreTrainedModel
,
)
from
.models.mt5
import
MT5EncoderModel
,
MT5ForConditionalGeneration
,
MT5Model
from
.models.mt5
import
MT5EncoderModel
,
MT5ForConditionalGeneration
,
MT5Model
,
MT5PreTrainedModel
from
.models.mvp
import
(
MVP_PRETRAINED_MODEL_ARCHIVE_LIST
,
MvpForCausalLM
,
...
...
src/transformers/models/mt5/__init__.py
View file @
15e17c99
...
...
@@ -51,7 +51,13 @@ try:
except
OptionalDependencyNotAvailable
:
pass
else
:
_import_structure
[
"modeling_mt5"
]
=
[
"MT5EncoderModel"
,
"MT5ForConditionalGeneration"
,
"MT5Model"
]
_import_structure
[
"modeling_mt5"
]
=
[
"MT5EncoderModel"
,
"MT5ForConditionalGeneration"
,
"MT5Model"
,
"MT5PreTrainedModel"
,
"MT5Stack"
,
]
try
:
if
not
is_tf_available
():
...
...
@@ -79,7 +85,7 @@ if TYPE_CHECKING:
except
OptionalDependencyNotAvailable
:
pass
else
:
from
.modeling_mt5
import
MT5EncoderModel
,
MT5ForConditionalGeneration
,
MT5Model
from
.modeling_mt5
import
MT5EncoderModel
,
MT5ForConditionalGeneration
,
MT5Model
,
MT5PreTrainedModel
,
MT5Stack
try
:
if
not
is_tf_available
():
...
...
src/transformers/models/mt5/modeling_mt5.py
View file @
15e17c99
This diff is collapsed.
Click to expand it.
src/transformers/utils/dummy_pt_objects.py
View file @
15e17c99
...
...
@@ -4003,6 +4003,13 @@ class MT5Model(metaclass=DummyObject):
requires_backends
(
self
,
[
"torch"
])
class
MT5PreTrainedModel
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
])
MVP_PRETRAINED_MODEL_ARCHIVE_LIST
=
None
...
...
utils/check_repo.py
View file @
15e17c99
...
...
@@ -40,6 +40,7 @@ PRIVATE_MODELS = [
"LongT5Stack"
,
"RealmBertModel"
,
"T5Stack"
,
"MT5Stack"
,
"SwitchTransformersStack"
,
"TFDPRSpanPredictor"
,
"MaskFormerSwinModel"
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment