fix #2649

Former-commit-id: 1c850de660c671d92f0bc63f230d338b60b7c0bd
2024-03-01 13:02:41 +08:00
parent 5306a71b42
commit 59a9a5994e
9 changed files with 99 additions and 87 deletions
--- a/src/llmtuner/model/init.py
+++ b/src/llmtuner/model/init.py
@@ -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",
+]