carry/gzhu-biyesheji
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tools): 修正__init__.py使得model.py正确导入

Browse Source
This commit is contained in:
carry 2025-04-12 01:16:07 +08:00
parent 611904cef9
commit 9784f2aed3
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/__init__.py
View File

@ -1,4 +1,4 @@
from .parse_markdown import parse_markdown from .parse_markdown import parse_markdown
from .scan_doc_dir import * from .scan_doc_dir import *
from .json_example import generate_example_json from .json_example import generate_example_json
from .model import get_model_name from .model import *

1
tools/model.py
View File

@ -9,7 +9,6 @@ def formatting_prompts_func(examples,tokenizer):
questions = examples["question"] questions = examples["question"]
answer = examples["answer"] answer = examples["answer"]
# 将Question和Response组合成对话形式
convos = [ convos = [
[{"role": "user", "content": q}, {"role": "assistant", "content": r}] [{"role": "user", "content": q}, {"role": "assistant", "content": r}]
for q, r in zip(questions, answer) for q, r in zip(questions, answer)