update default path

This commit is contained in:
JustinLin610
2023-08-27 18:47:09 +08:00
parent 50131acda4
commit 6703410770

View File

@@ -14,7 +14,7 @@ import torch
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig from transformers.generation import GenerationConfig
DEFAULT_CKPT_PATH = 'QWen/QWen-7B-Chat' DEFAULT_CKPT_PATH = 'Qwen/Qwen-7B-Chat'
def _get_args(): def _get_args():