Files
llm_trainer/src/llamafactory/model/__init__.py
2024-05-16 18:39:08 +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",
]