0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pref: reduce GC events

Browse source
This commit is contained in:
ashilkn 2024-03-27 16:34:06 +05:30
parent 0fc3773812
commit 35a8a96898
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
mobile/lib/services/machine_learning/semantic_search/frameworks/onnx/onnx_image_encoder.dart
View file

@ -15,9 +15,10 @@ class OnnxImageEncoder {
..setIntraOpNumThreads(1)
..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
try {
final bytes = await File(args["imageModelPath"]).readAsBytes();
final session = OrtSession.fromBuffer(bytes, sessionOptions);
final session =
OrtSession.fromFile(File(args["imageModelPath"]), sessionOptions);
_logger.info('image model loaded');
return session.address;
} catch (e, s) {
_logger.severe(e, s);