merge model part to the text stream

Former-commit-id: b6fcb832ddaed4647d6f2b926f3dfccd47f3ea84
2024-04-25 08:20:41 +08:00
parent 5142349661
commit 00e2a272ef
5 changed files with 24 additions and 172 deletions
--- a/src/llmtuner/model/init.py
+++ b/src/llmtuner/model/init.py
@@ -1,10 +1,9 @@
-from .loader import load_config, load_model, load_tokenizer, load_mm_model
+from .loader import load_config, load_model, load_tokenizer, load_processor
 from .utils.misc import find_all_linear_modules, load_valuehead_params

 __all__ = [
    "load_config",
    "load_model",
-    "load_mm_model",
    "load_tokenizer",
    "load_processor",
    "load_valuehead_params",