423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 03:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #2649

Browse Source
This commit is contained in:
hiyouga
2024-03-01 13:02:41 +08:00
parent 396fd47947
commit 4e5fae2fac
9 changed files with 99 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/llmtuner/model/__init__.py
View File

@@ -1,5 +1,11 @@
from .loader import load_model_and_tokenizer
from .loader import load_model, load_model_and_tokenizer, load_tokenizer
from .utils import dispatch_model, load_valuehead_params
__all__ = ["load_model_and_tokenizer", "dispatch_model", "load_valuehead_params"]
__all__ = [
"load_model",
"load_model_and_tokenizer",
"load_tokenizer",
"dispatch_model",
"load_valuehead_params",
]