Files
llm_trainer/src/llmtuner/model/__init__.py
2023-11-16 02:27:03 +08:00

6 lines
274 B
Python

# Level: loader > adapter > parser, utils
from llmtuner.model.loader import load_model_and_tokenizer
from llmtuner.model.parser import get_train_args, get_infer_args, get_eval_args
from llmtuner.model.utils import dispatch_model, generate_model_card, load_valuehead_params