423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 20:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix import bug

Browse Source
This commit is contained in:
hiyouga
2023-11-16 02:27:03 +08:00
parent ce78303600
commit 35b91ea34c
6 changed files with 91 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/model/__init__.py
View File

@@ -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