fix #3347 #3387

Former-commit-id: c253c18185a29b59190f3e0ed236c2bb4c788085
2024-04-24 01:30:16 +08:00
parent 1e4010a1fb
commit 35c4a2c212
3 changed files with 31 additions and 13 deletions
--- a/src/llmtuner/model/init.py
+++ b/src/llmtuner/model/init.py
@@ -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",