Update modeling_ltgbert.py

modeling_ltgbert.py

@@ -24,7 +24,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 from torch.utils import checkpoint
 
-from configuration_ltgbert import LtgBertConfig
+from .configuration_ltgbert import LtgBertConfig
 from transformers.modeling_utils import PreTrainedModel
 from transformers.activations import gelu_new
 from transformers.modeling_outputs import (
@@ -39,34 +39,10 @@ from transformers.pytorch_utils import softmax_backward_data
 from transformers.utils import add_start_docstrings, add_start_docstrings_to_model_forward
 
 
-_CHECKPOINT_FOR_DOC = "ltg/bnc-bert-span"
+_CHECKPOINT_FOR_DOC = "ltg/ltg-bert-bnc"
 _CONFIG_FOR_DOC = "LtgBertConfig"
 
 
-LTG_BERT_PRETRAINED_MODEL_ARCHIVE_LIST = [
-    "bnc-bert-span",
-    "bnc-bert-span-2x",
-    "bnc-bert-span-0.5x",
-    "bnc-bert-span-0.25x",
-    "bnc-bert-span-order",
-    "bnc-bert-span-document",
-    "bnc-bert-span-word",
-    "bnc-bert-span-subword",
-
-    "norbert3-xs",
-    "norbert3-small",
-    "norbert3-base",
-    "norbert3-large",
-
-    "norbert3-oversampled-base",
-    "norbert3-ncc-base",
-    "norbert3-nak-base",
-    "norbert3-nb-base",
-    "norbert3-wiki-base",
-    "norbert3-c4-base"
-]
-
-
 class Encoder(nn.Module):
     def __init__(self, config, activation_checkpointing=False):
         super().__init__()