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
OpenDAS
Megatron-LM
Commits
da1c96e9
"examples/vscode:/vscode.git/clone" did not exist on "ff43dba7eaee6d2055d299cf58183b8d19a35daa"
Commit
da1c96e9
authored
Jul 30, 2021
by
Deepak Narayanan
Browse files
Add assertion for now preventing usage of pipeline parallelism with T5 model
parent
2babcaf6
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
megatron/model/language_model.py
megatron/model/language_model.py
+4
-0
No files found.
megatron/model/language_model.py
View file @
da1c96e9
...
@@ -331,6 +331,10 @@ class TransformerLanguageModel(MegatronModule):
...
@@ -331,6 +331,10 @@ class TransformerLanguageModel(MegatronModule):
# Decoder (usually set to False, True if part of an encoder-decoder
# Decoder (usually set to False, True if part of an encoder-decoder
# architecture and in decoder-only stage).
# architecture and in decoder-only stage).
if
self
.
add_decoder
:
if
self
.
add_decoder
:
# Temporary assertion until we verify correctness of pipeline parallelism
# implementation of T5.
assert
args
.
pipeline_model_parallel_size
==
1
,
\
'pipeline parallelism is not supported in the presence of decoder'
self
.
decoder
=
ParallelTransformer
(
self
.
decoder
=
ParallelTransformer
(
self
.
init_method
,
self
.
init_method
,
output_layer_init_method
,
output_layer_init_method
,
...
...
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