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

Use the file directly to create an OrtSession

Browse source
This commit is contained in:
vishnukvmd 2023-12-19 00:35:16 +05:30
parent 28d165ad08
commit 836fb97598
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/services/semantic_search/frameworks/onnx/onnx_text_encoder.dart
View file

@ -31,8 +31,8 @@ class OnnxTextEncoder {
..setSessionGraphOptimizationLevel(GraphOptimizationLevel.ortEnableAll);
try {
_logger.info("Loading text model");
final bytes = File(args["textModelPath"]).readAsBytesSync();
final session = OrtSession.fromBuffer(bytes, sessionOptions);
final session =
OrtSession.fromFile(File(args["textModelPath"]), sessionOptions);
_logger.info('text model loaded');
return session.address;
} catch (e, s) {