update default path

main
JustinLin610 2 years ago
parent 5510223978
commit 09f189449c

@ -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'

Loading…
Cancel
Save