feat(example, ios): use metal by default (disable coreml)

mybigday · Nov 11, 2024 · b0408b0 · b0408b0
1 parent 6229cf5
commit b0408b0
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/example/src/context-opts.ios.ts b/example/src/context-opts.ios.ts
@@ -1,19 +1,21 @@
 import { Platform } from 'react-native'
 
+const getCoreMLModelAsset = () =>
+  Platform.OS === 'ios'
+    ? {
+        filename: 'ggml-base-encoder.mlmodelc',
+        assets: [
+          require('../assets/ggml-base-encoder.mlmodelc/weights/weight.bin'),
+          require('../assets/ggml-base-encoder.mlmodelc/model.mil'),
+          require('../assets/ggml-base-encoder.mlmodelc/coremldata.bin'),
+        ],
+      }
+    : undefined
+
 export default {
-  useCoreMLIos: true,
-  // If you don't want to enable Core ML, you can remove this property
-  coreMLModelAsset:
-    Platform.OS === 'ios'
-      ? {
-          filename: 'ggml-base-encoder.mlmodelc',
-          assets: [
-            require('../assets/ggml-base-encoder.mlmodelc/weights/weight.bin'),
-            require('../assets/ggml-base-encoder.mlmodelc/model.mil'),
-            require('../assets/ggml-base-encoder.mlmodelc/coremldata.bin'),
-          ],
-        }
-      : undefined,
-  useGpu: false, // Enable Metal (Will skip Core ML if enabled)
-  useFlashAttn: false,
+  useGpu: true, // Enable Metal (Will skip Core ML if enabled)
+  useFlashAttn: true,
+
+  useCoreMLIos: false,
+  coreMLModelAsset: getCoreMLModelAsset(),
 }