p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: fix bug

This commit is contained in:
wql 2024-09-05 13:04:54 +08:00
parent cc99691cf4
commit ceb01459fe
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prepare_yaml_file.py
View File

@ -12,7 +12,9 @@ def main():
yaml_file = './results/lora_sft_template.yaml' yaml_file = './results/lora_sft_template.yaml'
elif run_type == "inference": elif run_type == "inference":
yaml_file = './results/predict_template.yaml' yaml_file = './results/predict_template.yaml'
model_name_or_path = ""
template = ""
if model == "9g-8B": if model == "9g-8B":
model_name_or_path = "../../models/sft_8b_v2" model_name_or_path = "../../models/sft_8b_v2"
template = "" template = ""