Update modeling_TAAS.py

modeling_TAAS.py

@@ -8,18 +8,18 @@ from torch.nn import Parameter
 from torch.nn.init import normal_
 import torch.utils.checkpoint
 from torch import Tensor, device
-from TAAS_utils import * 
+from .TAAS_utils import * 
 from transformers.modeling_utils import ModuleUtilsMixin
 from transformers import AutoTokenizer, AutoModel, BertTokenizer
-from graphormer import Graphormer3D 
+from .graphormer import Graphormer3D 
 import pickle
 import torch
 import sys
-from ner_model import NER_model
+from .ner_model import NER_model
 import numpy as np
 
 
-from htc_loss import HTCLoss
+from .htc_loss import HTCLoss
 from transformers.utils.hub import cached_file
 remap_code_2_chn_file_path = cached_file( 
     'Cainiao-AI/TAAS',