modity code structure

Former-commit-id: 0682ed357210897e0b67c4a6eb31a94b3eb929f1
This commit is contained in:
hiyouga
2023-07-15 16:54:28 +08:00
parent fa06b168ab
commit 6261fb362a
57 changed files with 1999 additions and 1816 deletions

7
src/llmtuner/__init__.py Normal file
View File

@@ -0,0 +1,7 @@
from llmtuner.api import create_app
from llmtuner.extras.misc import get_logits_processor
from llmtuner.extras.template import Template
from llmtuner.tuner import get_train_args, get_infer_args, load_model_and_tokenizer, run_pt, run_sft, run_rm, run_ppo
__version__ = "0.0.9"