Update modeling_nt_bert.py
Browse files- modeling_nt_bert.py +1 -1
modeling_nt_bert.py
CHANGED
@@ -7,7 +7,7 @@ from einops import rearrange
|
|
7 |
from huggingface_hub import hf_hub_download
|
8 |
from mup import MuReadout, set_base_shapes
|
9 |
from mup.init import normal_
|
10 |
-
from
|
11 |
from rotary_embedding_torch import RotaryEmbedding
|
12 |
from transformers.modeling_outputs import (
|
13 |
BaseModelOutputWithPastAndCrossAttentions,
|
|
|
7 |
from huggingface_hub import hf_hub_download
|
8 |
from mup import MuReadout, set_base_shapes
|
9 |
from mup.init import normal_
|
10 |
+
from .configuring_nt_bert import BertConfig
|
11 |
from rotary_embedding_torch import RotaryEmbedding
|
12 |
from transformers.modeling_outputs import (
|
13 |
BaseModelOutputWithPastAndCrossAttentions,
|