From ff7a615db774b233416e71d2a7fdd9b7478d7827 Mon Sep 17 00:00:00 2001
From: David Koski <dkoski@apple.com>
Date: Fri, 1 Mar 2024 22:45:01 -0800
Subject: [PATCH] improve phi prompt -- partial fix for #9

---
 Libraries/LLM/Models.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Libraries/LLM/Models.swift b/Libraries/LLM/Models.swift
index 7d390b6..a478e43 100644
--- a/Libraries/LLM/Models.swift
+++ b/Libraries/LLM/Models.swift
@@ -70,7 +70,7 @@ extension ModelConfiguration {
 
     public static let phi4bit = ModelConfiguration(id: "mlx-community/phi-2-hf-4bit-mlx") {
         prompt in
-        "Instruct: \(prompt). Output: "
+        "Instruct: \(prompt)\nOutput: "
     }
 
     public static let gemma2bQuantized = ModelConfiguration(