Files
LLaMA-Factory/src/llmtuner/model/__init__.py
2024-03-13 23:32:51 +08:00

12 lines
287 B
Python

from .loader import load_model, load_model_and_tokenizer, load_tokenizer
from .utils import find_all_linear_modules, load_valuehead_params
__all__ = [
"load_model",
"load_model_and_tokenizer",
"load_tokenizer",
"load_valuehead_params",
"find_all_linear_modules",
]