This commit is contained in:
hiyouga
2024-04-24 01:30:16 +08:00
parent 1a13f05555
commit 707f0b1d5d
3 changed files with 31 additions and 13 deletions

View File

@@ -1,8 +1,9 @@
from .loader import load_model, load_tokenizer
from .loader import load_config, load_model, load_tokenizer
from .utils import find_all_linear_modules, load_valuehead_params
__all__ = [
"load_config",
"load_model",
"load_tokenizer",
"load_valuehead_params",