[mob][photos] Small fix

This commit is contained in:
laurenspriem
2024-06-29 19:17:47 +05:30
parent dcb90f9e59
commit 5eb02c242c
2 changed files with 6 additions and 16 deletions

View File

@@ -54,14 +54,9 @@ class ONNX extends MLFramework {
@override
Future<void> loadTextModel(String path) async {
_logger.info('loadTextModel called');
final startTime = DateTime.now();
final String vocabPath = await _clipText.getVocab();
await _computer.compute(
_clipText.initTokenizer,
param: {
"vocabPath": vocabPath,
},
);
await _clipText.initTokenizer();
_textEncoderAddress = await _computer.compute(
_clipText.loadModel,
param: {

View File

@@ -1,7 +1,7 @@
import "dart:io";
import "dart:math";
import "dart:typed_data";
import "package:flutter/foundation.dart";
import "package:logging/logging.dart";
import "package:onnxruntime/onnxruntime.dart";
import 'package:photos/services/machine_learning/semantic_search/frameworks/onnx/onnx_text_tokenizer.dart';
@@ -11,18 +11,13 @@ class OnnxTextEncoder {
static const kVocabRemotePath =
"https://models.ente.io/bpe_simple_vocab_16e6.txt";
final _logger = Logger("OnnxTextEncoder");
final OnnxTextTokenizer _tokenizer = OnnxTextTokenizer();
Future<String> getVocab() async {
Future<void> initTokenizer() async {
final File vocabFile =
await RemoteAssetsService.instance.getAsset(kVocabRemotePath);
return vocabFile.path;
}
// Do not run in an isolate since rootBundle can only be accessed in the main isolate
Future<void> initTokenizer(Map args) async {
final String path = args["vocabPath"];
final File vocabFile = File(path);
final String vocab = await vocabFile.readAsString();
await _tokenizer.init(vocab);
}