fix: fix model name

This commit is contained in:
wql 2024-09-25 10:21:25 +08:00
parent 78b6f5b60c
commit 379f6d1c47
2 changed files with 9 additions and 9 deletions

View File

@ -25,14 +25,14 @@ for ((i=0; i<repeat_cnt; i++)); do
bash run_once.sh inference Baichuan2-7B 1 500 bash run_once.sh inference Baichuan2-7B 1 500
bash run_once.sh inference Baichuan2-7B 1 1000 bash run_once.sh inference Baichuan2-7B 1 1000
bash run_once.sh lora_sft chatglm2-6B 8 500 bash run_once.sh lora_sft ChatGLM2-6B 8 500
bash run_once.sh lora_sft chatglm2-6B 8 1000 bash run_once.sh lora_sft ChatGLM2-6B 8 1000
bash run_once.sh lora_sft chatglm2-6B 1 500 bash run_once.sh lora_sft ChatGLM2-6B 1 500s
bash run_once.sh lora_sft chatglm2-6B 1 1000 bash run_once.sh lora_sft ChatGLM2-6B 1 1000
bash run_once.sh inference chatglm2-6B 8 500 bash run_once.sh inference ChatGLM2-6B 8 500
bash run_once.sh inference chatglm2-6B 8 1000 bash run_once.sh inference ChatGLM2-6B 8 1000
bash run_once.sh inference chatglm2-6B 1 500 bash run_once.sh inference ChatGLM2-6B 1 500
bash run_once.sh inference chatglm2-6B 1 1000 bash run_once.sh inference ChatGLM2-6B 1 1000
# bash run_once.sh lora_sft llama-2-7b-ms 8 500 # bash run_once.sh lora_sft llama-2-7b-ms 8 500
# bash run_once.sh lora_sft llama-2-7b-ms 8 1000 # bash run_once.sh lora_sft llama-2-7b-ms 8 1000

View File

@ -31,7 +31,7 @@ def main():
model_name_or_path = "../../../models/qwen" model_name_or_path = "../../../models/qwen"
template = "qwen" template = "qwen"
else: else:
print("ERROR: model not supported.") print("ERROR: model not supported or model name wrong")
sys.exit() sys.exit()
config = None config = None