Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
megatron/model/__init__.py
View file @ 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