add some

Former-commit-id: 8d035a849c4a441d457791aab073861adf69a09f
2024-04-25 21:08:32 +08:00
parent 9b210cf4b3
commit dbd905438b
8 changed files with 80 additions and 283 deletions
--- a/src/llmtuner/model/init.py
+++ b/src/llmtuner/model/init.py
@@ -1,11 +1,10 @@
-from .loader import load_config, load_model, load_tokenizer, load_processor
+from .loader import load_config, load_model, load_tokenizer
 from .utils.misc import find_all_linear_modules, load_valuehead_params

 __all__ = [
    "load_config",
    "load_model",
    "load_tokenizer",
-    "load_processor",
    "load_valuehead_params",
    "find_all_linear_modules",
 ]