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,
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
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