Files
LLaMA-Factory/src/llmtuner/model/__init__.py
2024-04-28 03:01:49 +08:00

13 lines
294 B
Python

from .loader import load_config, load_model, load_tokenizer
from .utils.misc import find_all_linear_modules
from .utils.valuehead import load_valuehead_params
__all__ = [
"load_config",
"load_model",
"load_tokenizer",
"load_valuehead_params",
"find_all_linear_modules",
]