From a49a63316489d6441e84678a44d7db2fd1581ebd Mon Sep 17 00:00:00 2001 From: vfbd Date: Mon, 22 Aug 2022 16:46:39 -0400 Subject: [PATCH] `self.ckpt_path` -> `self.data.ckpt_path` --- prompt_tuner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/prompt_tuner.py b/prompt_tuner.py index f172c6a1..2fbd6ee2 100644 --- a/prompt_tuner.py +++ b/prompt_tuner.py @@ -251,7 +251,7 @@ class TrainerBase(abc.ABC): return True def get_tokenizer(self) -> transformers.PreTrainedTokenizerBase: - return get_tokenizer(self.ckpt_path) + return get_tokenizer(self.data.ckpt_path) def save_data(self): pass