From 9784f2aed371ee64151e26f15ee8429b8769aa86 Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Sat, 12 Apr 2025 01:16:07 +0800 Subject: [PATCH] =?UTF-8?q?fix(tools):=20=E4=BF=AE=E6=AD=A3=5F=5Finit=5F?= =?UTF-8?q?=5F.py=E4=BD=BF=E5=BE=97model.py=E6=AD=A3=E7=A1=AE=E5=AF=BC?= =?UTF-8?q?=E5=85=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/__init__.py | 2 +- tools/model.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) 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)