fix import bug

Former-commit-id: 2356029cdd120d5f7bf630b80681ce8c53bff90d
2023-11-16 02:27:03 +08:00
parent 7a3a0144a5
commit f81a8a5e5c
6 changed files with 91 additions and 84 deletions
--- a/src/llmtuner/model/init.py
+++ b/src/llmtuner/model/init.py
@@ -1,3 +1,5 @@
+# Level: loader > adapter > parser, utils
+
 from llmtuner.model.loader import load_model_and_tokenizer
 from llmtuner.model.parser import get_train_args, get_infer_args, get_eval_args
-from llmtuner.model.utils import create_ref_model, create_reward_model, dispatch_model, generate_model_card
+from llmtuner.model.utils import dispatch_model, generate_model_card, load_valuehead_params