update default path

main
JustinLin610 1 year ago
parent 50131acda4
commit 6703410770

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

Loading…
Cancel
Save