ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix #1558

Browse Source 
Former-commit-id: 263b2b24c8a649b51fa5ae768a24e67def8e0e96
This commit is contained in:
hiyouga
2023-11-19 14:15:47 +08:00
parent 3d1ee27ccd
commit 6889f044fb
8 changed files with 35 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,4 +2,4 @@
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 dispatch_model, generate_model_card, load_valuehead_params
from llmtuner.model.utils import dispatch_model, get_modelcard_args, load_valuehead_params

2
src/llmtuner/model/utils.py
View File

@@ -71,7 +71,7 @@ def find_all_linear_modules(
return list(module_names)
def generate_model_card(
def get_modelcard_args(
model_args: "ModelArguments",
data_args: "DataArguments",
finetuning_args: "FinetuningArguments"