update default path

This commit is contained in:
JustinLin610
2023-08-27 18:49:11 +08:00
parent 5510223978
commit 09f189449c

View File

@@ -14,6 +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'