Commit 8cd16667 authored by Mohammad Shoeybi's avatar Mohammad Shoeybi
Browse files

Merge branch 'no-realm-import' into 'main' 

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

See merge request ADLR/megatron-lm!241
parents 77d54fbb c6f18ccf

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