improve phi prompt -- partial fix for #9

2024-03-01 22:45:01 -08:00
parent 23fc53c43e
commit ff7a615db7
1 changed files with 1 additions and 1 deletions
--- 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(