423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 03:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

create chat model

Browse Source
This commit is contained in:
hiyouga
2023-07-15 19:26:20 +08:00
parent d640c5545f
commit 657cf0f55a
8 changed files with 117 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/__init__.py
View File

@@ -1,6 +1,5 @@
from llmtuner.api import create_app
from llmtuner.extras.misc import get_logits_processor
from llmtuner.extras.template import Template
from llmtuner.chat import ChatModel
from llmtuner.tuner import get_train_args, get_infer_args, load_model_and_tokenizer, run_pt, run_sft, run_rm, run_ppo