`self.ckpt_path` -> `self.data.ckpt_path`

2022-08-22 16:46:39 -04:00 · 2022-08-22 16:46:39 -04:00 · a49a633164
parent 05cf9b1dde
commit a49a633164
1 changed files with 1 additions and 1 deletions
--- 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