From 3cc05d5c420bde52f7b5efff30b7a034ee1f1975 Mon Sep 17 00:00:00 2001 From: laurenspriem Date: Fri, 26 Jul 2024 17:39:34 +0200 Subject: [PATCH] [mob][photos] Fix in isolate message --- mobile/lib/services/machine_learning/ml_computer.dart | 6 +++--- .../lib/services/machine_learning/ml_indexing_isolate.dart | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mobile/lib/services/machine_learning/ml_computer.dart b/mobile/lib/services/machine_learning/ml_computer.dart index 9c4fabdd4f..753725abc0 100644 --- a/mobile/lib/services/machine_learning/ml_computer.dart +++ b/mobile/lib/services/machine_learning/ml_computer.dart @@ -21,7 +21,7 @@ enum MLComputerOperation { } class MLComputer { - final _logger = Logger('MLComputerIsolate'); + final _logger = Logger('MLComputer'); final _initLock = Lock(); final _functionLock = Lock(); @@ -99,7 +99,7 @@ class MLComputer { break; case MLComputerOperation.runClipText: final textEmbedding = await ClipTextEncoder.predict(args); - sendPort.send(List.from(textEmbedding, growable: false)); + sendPort.send(List.from(textEmbedding, growable: false)); break; } } catch (e, stackTrace) { @@ -158,8 +158,8 @@ class MLComputer { } Future> runClipText(String query) async { + await _ensureLoadedClipTextModel(); try { - await _ensureLoadedClipTextModel(); final int clipAddress = ClipTextEncoder.instance.sessionAddress; final textEmbedding = await _runInIsolate( ( diff --git a/mobile/lib/services/machine_learning/ml_indexing_isolate.dart b/mobile/lib/services/machine_learning/ml_indexing_isolate.dart index 481dc7915c..7636135289 100644 --- a/mobile/lib/services/machine_learning/ml_indexing_isolate.dart +++ b/mobile/lib/services/machine_learning/ml_indexing_isolate.dart @@ -99,7 +99,7 @@ class MLIndexingIsolate { ); addresses.add(address); } - sendPort.send(List.from(addresses, growable: false)); + sendPort.send(List.from(addresses, growable: false)); break; case MLIndexingOperation.releaseModels: final modelNames = args['modelNames'] as List;