Upload New File

f78e0a30 · zhougaofeng · dc330232 · f78e0a30
Commit f78e0a30 authored Jun 11, 2024 by zhougaofeng
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 0 deletions

src/llmfactory/model/__init__.py src/llmfactory/model/__init__.py +12 -0

No files found.
--- a/src/llmfactory/model/__init__.py
+++ b/src/llmfactory/model/__init__.py
+from .loader import load_config, load_model, load_tokenizer
+from .utils.misc import find_all_linear_modules
+from .utils.valuehead import load_valuehead_params
+
+
+__all__ = [
+    "load_config",
+    "load_model",
+    "load_tokenizer",
+    "load_valuehead_params",
+    "find_all_linear_modules",
+]