diff --git a/mobile/lib/services/machine_learning/semantic_search/clip/clip_image_encoder.dart b/mobile/lib/services/machine_learning/semantic_search/clip/clip_image_encoder.dart index ac1d9e20da..1b6e8a9ebe 100644 --- a/mobile/lib/services/machine_learning/semantic_search/clip/clip_image_encoder.dart +++ b/mobile/lib/services/machine_learning/semantic_search/clip/clip_image_encoder.dart @@ -55,7 +55,9 @@ class ClipImageEncoder extends MlModel { final embedding = (outputs[0]?.value as List>)[0]; inputOrt.release(); runOptions.release(); - outputs.forEach((element) => element?.release()); + for (var element in outputs) { + element?.release(); + } normalizeEmbedding(embedding); w.stopWithLog("done"); return embedding; diff --git a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart index a7c846c29c..3c06bfc31d 100644 --- a/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart +++ b/mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart @@ -303,6 +303,7 @@ class SemanticSearchService { ByteData imageByteData, int clipImageAddress, ) async { + final startTime = DateTime.now(); final embedding = await ClipImageEncoder.predict( image, imageByteData, @@ -311,6 +312,9 @@ class SemanticSearchService { final clipResult = ClipResult(fileID: enteFileID, embedding: embedding); + dev.log('Finished running ClipImage for $enteFileID in ' + '${DateTime.now().difference(startTime).inMilliseconds} ms'); + return clipResult; } }