TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][photos] Rename for consistency

Browse Source
This commit is contained in:
laurenspriem
2024-07-23 11:48:14 +02:00
parent b8ccc74dac
commit 56b4d3cd46
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mobile/lib/services/machine_learning/semantic_search/clip/clip_text_encoder.dart
View File

@@ -30,11 +30,12 @@ class ClipTextEncoder extends MlModel {
static final instance = ClipTextEncoder._privateConstructor();
factory ClipTextEncoder() => instance;
static Future<List<double>> infer(Map args) async {
static Future<List<double>> predict(Map args) async {
final text = args["text"];
final address = args["address"] as int;
final vocabPath = args["vocabPath"] as String;
final List<int> tokenize = await ClipTextTokenizer.instance.tokenize(text, vocabPath);
final List<int> tokenize =
await ClipTextTokenizer.instance.tokenize(text, vocabPath);
final int32list = Int32List.fromList(tokenize);
return _runFFIBasedPredict(int32list, address);
}

2
mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
View File

@@ -302,7 +302,7 @@ class SemanticSearchService {
final String tokenizerVocabPath =
await RemoteAssetsService.instance.getAssetPath(remotePath);
final textEmbedding = await _computer.compute(
ClipTextEncoder.infer,
ClipTextEncoder.predict,
param: {
"text": query,
"address": clipAddress,