modity code structure

This commit is contained in:
hiyouga
2023-07-15 16:54:28 +08:00
parent 2a0f1f8398
commit f751376613
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"