Add whitespace after "Instruct:" (#1277)

Lightning-AI · Apr 12, 2024 · d6e91ee · d6e91ee
1 parent 3516bea
commit d6e91ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litgpt/prompts.py b/litgpt/prompts.py
@@ -251,7 +251,7 @@ def stop_tokens(self, tokenizer: "Tokenizer") -> Tuple[List[int], ...]:
 
 class Phi2(PromptStyle):
     def apply(self, prompt: str, **kwargs: str) -> str:
-        return f"Instruct:{prompt}\nOutput:"
+        return f"Instruct: {prompt}\nOutput:"
 
 
 class TinyLlama(PromptStyle):