🎨 Format Code

johnmai-dev · Sep 21, 2024 · 41f42b7 · 41f42b7
1 parent 3b1752e
commit 41f42b7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
diff --git a/ChatMLX/Components/UltramanNavigationSplitView.swift b/ChatMLX/Components/UltramanNavigationSplitView.swift
@@ -69,9 +69,11 @@ extension View {
         preference(
             key: UltramanNavigationToolbarKey.self,
             value: [
-                UltramanToolbarItem(alignment: alignment, content: {
-                    content()
-                })
+                UltramanToolbarItem(
+                    alignment: alignment,
+                    content: {
+                        content()
+                    })
             ]
         )
     }

diff --git a/ChatMLX/Utilities/LLMRunner.swift b/ChatMLX/Utilities/LLMRunner.swift
@@ -6,10 +6,10 @@
 //
 
 import Defaults
-import Metal
 import MLX
 import MLXLLM
 import MLXRandom
+import Metal
 import SwiftUI
 import Tokenizers
 
@@ -117,10 +117,8 @@ class LLMRunner {
 
                 let result = await modelContainer.perform {
                     model,
-                        tokenizer in
-
-
-
+                    tokenizer in
+
                     MLXLLM.generate(
                         promptTokens: messageTokens,
                         parameters: GenerateParameters(
@@ -132,7 +130,9 @@ class LLMRunner {
                         ),
                         model: model,
                         tokenizer: tokenizer,
-                        extraEOSTokens: modelConfiguration.extraEOSTokens.union(["<|im_end|>", "<|end|>"])
+                        extraEOSTokens: modelConfiguration.extraEOSTokens.union([
+                            "<|im_end|>", "<|end|>",
+                        ])
                     ) { tokens in
                         if tokens.count % displayEveryNTokens == 0 {
                             let text = tokenizer.decode(tokens: tokens)