From a1b454fc43fac59fa1b85cc80c5e7f4769c7928b Mon Sep 17 00:00:00 2001 From: bones-zhu <61820323+bones-zhu@users.noreply.github.com> Date: Fri, 21 Jul 2023 15:09:33 +0800 Subject: [PATCH] Update model_config.py fix type --- configs/model_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/model_config.py b/configs/model_config.py index ea24c36..dbb7bce 100644 --- a/configs/model_config.py +++ b/configs/model_config.py @@ -220,7 +220,7 @@ STREAMING = True # Use p-tuning-v2 PrefixEncoder USE_PTUNING_V2 = False -PTUNING_DIR='./ptuing-v2' +PTUNING_DIR='./ptuning-v2' # LLM running device LLM_DEVICE = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"