From 6b2ea75ca9648308d862e622ef4d3e61eb7979ff Mon Sep 17 00:00:00 2001
From: x54-729 <whxiaohao@163.com>
Date: Thu, 2 Nov 2023 22:05:54 +0800
Subject: [PATCH] fix import

---
 tools/transformers/convert2hf.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tools/transformers/convert2hf.py b/tools/transformers/convert2hf.py
index f8604df..7f3aa02 100644
--- a/tools/transformers/convert2hf.py
+++ b/tools/transformers/convert2hf.py
@@ -6,8 +6,7 @@ import re
 import tempfile
 
 import torch
-from modeling_internlm import InternLMConfig, InternLMForCausalLM
-from tokenization_internlm import InternLMTokenizer
+from internlm_model import InternLMConfig, InternLMForCausalLM,  InternLMTokenizer
 
 NUM_SHARDS = {
     "7B": 1,