fix(VisualRecognitionV3): temporary workaround for new A12 based devices

AnthonyOliveri · Nov 12, 2018 · 62edd09 · 62edd09
1 parent 938bfd2
commit 62edd09
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Source/VisualRecognitionV3/VisualRecognition+CoreML.swift b/Source/VisualRecognitionV3/VisualRecognition+CoreML.swift
@@ -279,7 +279,15 @@ extension VisualRecognition {
      */
     private func loadModelFromDisk(classifierID: String) throws -> MLModel {
         let modelURL = try locateModelOnDisk(classifierID: classifierID)
-        return try MLModel(contentsOf: modelURL)
+
+        // temporary workaround for compatibility issue with new A12 based devices
+        if #available(iOS 12.0, *) {
+            let modelConfig = MLModelConfiguration()
+            modelConfig.computeUnits = .cpuAndGPU
+            return try MLModel(contentsOf: modelURL, configuration: modelConfig)
+        } else {
+            return try MLModel(contentsOf: modelURL)
+        }
     }
 
     /// Convert results from Core ML classification requests into a `ClassifiedImages` model.