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

chore: modify model path

This commit is contained in:
wql 2024-09-13 11:17:15 +08:00
parent fac5e83af9
commit 6e698b5ce0
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
prepare_yaml_file.py
View File

@ -16,19 +16,19 @@ def main():
model_name_or_path = ""
template = ""
if model == "9g-8B":
model_name_or_path = "../../models/sft_8b_v2"
model_name_or_path = "ms_cache/hub/qy/sft_8b_v2"
template = "default"
elif model == "Baichuan2-7B":
model_name_or_path = "../../models/Baichuan2-7B-Base"
model_name_or_path = "ms_cache/hub/baichuan-inc/Baichuan2-7B-Base"
template = "baichuan2"
elif model == "ChatGLM2-6B":
model_name_or_path = "../../models/chatglm2-6b"
model_name_or_path = "ms_cache/hub/zhipuai/chatglm2-6b"
template = "chatglm2"
elif model == "Llama2-7B":
model_name_or_path = "../../models/llama-2-7b-ms"
model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms"
template = "llama2"
elif model == "Qwen-7B":
model_name_or_path = "../../models/Qwen-7B"
model_name_or_path = "ms_cache/hub/qwen/Qwen-7B"
template = "qwen"
else:
print("ERROR: model not supported.")

2
run_once.sh
View File

@ -46,6 +46,6 @@ fi
wait $train_pid
echo "Train ended"
sleep 60
# sleep 60
kill $npu_status_pid
echo "Npu status ended"