Commit c6f18ccf authored by Jared Casper's avatar Jared Casper
Browse files

Don't import deprecated model from realm_model which is broken.

parent 77d54fbb

megatron/model/__init__.py

+0 −2
Original line number Diff line number Diff line
@@ -34,13 +34,11 @@ from .bert_model import (BertModel, 
                         BertModelFirstStage,

                         BertModelIntermediateStage,

                         BertModelLastStage)

from .realm_model import ICTBertModel

from .gpt_model import (GPTModel,

                        GPTModelFirstStage,

                        GPTModelIntermediateStage,

                        GPTModelLastStage)

from .language_model import get_language_model

from .module import FP16Module

from .realm_model import ICTBertModel