Modification of infer.py and modeling_internlm.py

infer.py

@@ -9,8 +9,8 @@ from torch.utils.data import Dataset, DataLoader
 
 from transformers import get_linear_schedule_with_warmup, DataCollatorWithPadding
 
-from models.tokenization_internlm import InternLMTokenizer
-from models.modeling_internlm import InternLMForSequenceClassification
+from tokenization_internlm import InternLMTokenizer
+from modeling_internlm import InternLMForSequenceClassification
 
 
 logger = logging.getLogger(__name__)

modeling_internlm.py

@@ -42,7 +42,7 @@ try:
 except:  # noqa # pylint: disable=bare-except
     BaseStreamer = None
 
-from .configuration_internlm import InternLMConfig
+from configuration_internlm import InternLMConfig
 
 logger = logging.get_logger(__name__)