ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-01-31 06:42:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

create chat model

Browse Source 
Former-commit-id: bddf583b2fc099c957a1037418bd8504a837663e
This commit is contained in:
hiyouga
2023-07-15 19:26:20 +08:00
parent e9fe48150c
commit a8deee27f8
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