From 174b74ddc99acd2b541438826ad871ec16f740a9 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Tue, 9 May 2023 00:54:46 +0900 Subject: [PATCH] Rename variable to use same convention --- src/axolotl/prompt_tokenizers.py | 8 ++++---- src/axolotl/prompters.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/axolotl/prompt_tokenizers.py b/src/axolotl/prompt_tokenizers.py index 16dd7eea4..167648618 100644 --- a/src/axolotl/prompt_tokenizers.py +++ b/src/axolotl/prompt_tokenizers.py @@ -132,15 +132,15 @@ class CompletionPromptTokenizingStrategy(InstructionPromptTokenizingStrategy): ) def tokenize_prompt(self, prompt): - text = self.parse_instruction_fields(prompt) - full_prompt = self._build_full_prompt(text) + instruction = self.parse_instruction_fields(prompt) + full_prompt = self._build_full_prompt(instruction) tokenized_full_prompt = self._tokenize(full_prompt) return tokenized_full_prompt - def _build_full_prompt(self, text): + def _build_full_prompt(self, instruction): return self.prompter.build_prompt( - text + instruction ) diff --git a/src/axolotl/prompters.py b/src/axolotl/prompters.py index 920a50b78..914cbd0de 100644 --- a/src/axolotl/prompters.py +++ b/src/axolotl/prompters.py @@ -38,9 +38,9 @@ class JeopardyPrompter(AlpacaPrompter): class CompletionPrompter(AlpacaPrompter): def build_prompt( self, - text: str + instruction: str ) -> str: - return text + return instruction def get_response(self, output: str) -> str: return output.strip()