improve phi prompt -- partial fix for #9

ml-explore · Mar 2, 2024 · ff7a615 · ff7a615
1 parent 23fc53c
commit ff7a615
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(