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
...@@ -34,13 +34,11 @@ from .bert_model import (BertModel, ...@@ -34,13 +34,11 @@ from .bert_model import (BertModel,
BertModelFirstStage, BertModelFirstStage,
BertModelIntermediateStage, BertModelIntermediateStage,
BertModelLastStage) BertModelLastStage)
from .realm_model import ICTBertModel
from .gpt_model import (GPTModel, from .gpt_model import (GPTModel,
GPTModelFirstStage, GPTModelFirstStage,
GPTModelIntermediateStage, GPTModelIntermediateStage,
GPTModelLastStage) GPTModelLastStage)
from .language_model import get_language_model from .language_model import get_language_model
from .module import FP16Module from .module import FP16Module
from .realm_model import ICTBertModel
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment