chore: change model path

This commit is contained in:
wql 2024-09-18 15:26:05 +08:00
parent 4061a7899d
commit 9469b4c256
1 changed files with 1 additions and 1 deletions

View File

@ -28,7 +28,7 @@ def main():
model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms" model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms"
template = "llama2" template = "llama2"
elif model == "Qwen-7B": elif model == "Qwen-7B":
model_name_or_path = "ms_cache/hub/qwen/Qwen-7B" model_name_or_path = "../../../models/qwen"
template = "qwen" template = "qwen"
else: else:
print("ERROR: model not supported.") print("ERROR: model not supported.")