diff --git a/tools/__init__.py b/tools/__init__.py
index 901f13c..9fbd1c4 100644
--- a/tools/__init__.py
+++ b/tools/__init__.py
@@ -1,4 +1,4 @@
 from .parse_markdown import parse_markdown
 from .scan_doc_dir import *
 from .json_example import generate_example_json
-from .model import get_model_name
\ No newline at end of file
+from .model import *
\ No newline at end of file
diff --git a/tools/model.py b/tools/model.py
index 57b6bb6..860dacd 100644
--- a/tools/model.py
+++ b/tools/model.py
@@ -9,7 +9,6 @@ def formatting_prompts_func(examples,tokenizer):
     questions = examples["question"]
     answer = examples["answer"]
     
-    # 将Question和Response组合成对话形式
     convos = [
         [{"role": "user", "content": q}, {"role": "assistant", "content": r}]
         for q, r in zip(questions, answer)