Skip to content

Commit de5c2cb

Browse files
Myle Ottfacebook-github-bot
authored andcommitted
Fix model parallel LM (#1358)
Summary: Pull Request resolved: fairinternal/fairseq-py#1358 Reviewed By: alexeib Differential Revision: D24393064 Pulled By: myleott fbshipit-source-id: ee88fd1e7b203d7df6b7a65d3b1b1469e8fe9b6e
1 parent 9b8b464 commit de5c2cb

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

fairseq/model_parallel/models/transformer_lm.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@
2222

2323
@register_model("model_parallel_transformer_lm")
2424
class ModelParallelTransformerLanguageModel(TransformerLanguageModel):
25+
26+
@staticmethod
27+
def add_args(parser):
28+
TransformerLanguageModel.add_args(parser)
29+
2530
@classmethod
2631
def build_model(cls, args, task):
2732
"""Build a new model instance."""

0 commit comments

Comments
 (0)