meghanaraok commited on
Commit
ddf05a4
1 Parent(s): c142205

Update LongHiLATmain/models/modeling.py

Browse files
Files changed (1) hide show
  1. LongHiLATmain/models/modeling.py +1 -1
LongHiLATmain/models/modeling.py CHANGED
@@ -6,7 +6,7 @@ from torch.nn import BCEWithLogitsLoss, Dropout, Linear
6
  from transformers import AutoModel, XLNetModel, LongformerConfig
7
  from transformers.models.longformer.modeling_longformer import LongformerEncoder
8
  from huggingface_hub import PyTorchModelHubMixin
9
- from LongHiLATmain.hilat.models.utils import initial_code_title_vectors
10
 
11
  logger = logging.getLogger("lwat")
12
 
 
6
  from transformers import AutoModel, XLNetModel, LongformerConfig
7
  from transformers.models.longformer.modeling_longformer import LongformerEncoder
8
  from huggingface_hub import PyTorchModelHubMixin
9
+ from LongLAT.LongLAT.models.utils import initial_code_title_vectors
10
 
11
  logger = logging.getLogger("lwat")
12