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

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
8cd16667 · Mohammad Shoeybi · 77d54fbb · c6f18ccf · 8cd16667
Commit 8cd16667 authored Feb 25, 2021 by Mohammad Shoeybi
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 2 deletions

megatron/model/__init__.py megatron/model/__init__.py +0 -2

No files found.
--- a/megatron/model/__init__.py
+++ b/megatron/model/__init__.py
@@ -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