Add support for AceReason-Nemotron-7B (#321)

* support AceReason

Author: JoeJoe1313

Applications/MLXChatExample/Services/MLXService.swift

@@ -29,6 +29,7 @@ class MLXService {
             name: "qwen2.5VL:3b", configuration: VLMRegistry.qwen2_5VL3BInstruct4Bit, type: .vlm),
         LMModel(name: "qwen2VL:2b", configuration: VLMRegistry.qwen2VL2BInstruct4Bit, type: .vlm),
         LMModel(name: "smolVLM", configuration: VLMRegistry.smolvlminstruct4bit, type: .vlm),
+        LMModel(name: "acereason:7B", configuration: LLMRegistry.acereason_7b_4bit, type: .llm),
     ]
 
     /// Cache to store loaded model containers to avoid reloading.

Libraries/MLXLLM/Documentation.docc/Documentation.md

@@ -48,3 +48,4 @@ More advanced APIs are available for those that need them, see <doc:using-model>
 - ``Starcoder2Model``
 - ``MiMoModel``
 - ``GLM4Model``
+- ``AceReason``

Libraries/MLXLLM/LLMModelFactory.swift

@@ -45,6 +45,7 @@ public class LLMTypeRegistry: ModelTypeRegistry, @unchecked Sendable {
             "granite": create(GraniteConfiguration.self, GraniteModel.init),
             "mimo": create(MiMoConfiguration.self, MiMoModel.init),
             "glm4": create(GLM4Configuration.self, GLM4Model.init),
+            "acereason": create(Qwen2Configuration.self, Qwen2Model.init),
         ]
     }
 
@@ -211,6 +212,11 @@ public class LLMRegistry: AbstractModelRegistry, @unchecked Sendable {
         defaultPrompt: "Why is the sky blue?"
     )
 
+    static public let acereason_7b_4bit = ModelConfiguration(
+        id: "mlx-community/AceReason-Nemotron-7B-4bit",
+        defaultPrompt: ""
+    )
+
     private static func all() -> [ModelConfiguration] {
         [
             codeLlama13b4bit,
@@ -240,6 +246,7 @@ public class LLMRegistry: AbstractModelRegistry, @unchecked Sendable {
             smolLM_135M_4bit,
             mimo_7b_sft_4bit,
             glm4_9b_4bit,
+            acereason_7b_4bit,
         ]
     }
 

Libraries/MLXLLM/README.md

@@ -58,6 +58,7 @@ Currently supported model types are:
 - Starcoder2
 - MiMo
 - GLM4
+- AceReason
 
 See [llm-tool](../../Tools/llm-tool)