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