Clean up

Author: DePasqualeOrg

Libraries/MLXLLM/LLMModelFactory.swift

@@ -128,10 +128,7 @@ public class LLMRegistry: AbstractModelRegistry, @unchecked Sendable {
         id: "mlx-community/Phi-3.5-MoE-instruct-4bit",
         defaultPrompt: "What is the gravity on Mars and the moon?",
         extraEOSTokens: ["<|end|>"]
-    ) {
-        prompt in
-        "<|user|>\n\(prompt)<|end|>\n<|assistant|>\n"
-    }
+    )
 
     static public let gemma2bQuantized = ModelConfiguration(
         id: "mlx-community/quantized-gemma-2b-it",

Libraries/MLXLMCommon/ModelConfiguration.swift

@@ -24,7 +24,7 @@ public struct ModelConfiguration: Sendable {
         }
     }
 
-    /// pull the tokenizer from an alternate id
+    /// Alternate repo ID to use for the tokenizer
     public let tokenizerId: String?
 
     /// A reasonable default prompt for the model
@@ -72,7 +72,7 @@ public struct ModelConfiguration: Sendable {
     public func modelDirectory(hub: HubApi = HubApi()) -> URL {
         switch id {
         case .id(let id, _):
-            // download the model weights and config
+            // Download the model weights and config
             let repo = Hub.Repo(id: id)
             return hub.localRepoLocation(repo)