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

fix: fix file type

This commit is contained in:
wql 2024-09-05 13:03:22 +08:00
parent 4058fd7d64
commit cc99691cf4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prepare_yaml_file.py
View File

@ -9,9 +9,9 @@ def main():
output_dir = sys.argv[5]
if run_type == "lora_sft":
yaml_file = './results/lora_sft_template.yml'
yaml_file = './results/lora_sft_template.yaml'
elif run_type == "inference":
yaml_file = './results/predict_template.yml'
yaml_file = './results/predict_template.yaml'
if model == "9g-8B":
model_name_or_path = "../../models/sft_8b_v2"