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

Switch to GGML

Browse source
This commit is contained in:
vishnukvmd 2024-02-06 10:46:49 +05:30
parent 2edc63eb66
commit f0b4e71a51
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lib/services/semantic_search/semantic_search_service.dart
View file

@ -35,6 +35,7 @@ class SemanticSearchService {
static const kScoreThreshold = 0.23;
static const kShouldPushEmbeddings = true;
static const kCurrentModel = Model.onnxClip;
static const kIsGraphene = true;
static const kDebounceDuration = Duration(milliseconds: 4000);
final _logger = Logger("SemanticSearchService");
@ -77,7 +78,7 @@ class SemanticSearchService {
_hasInitialized = true;
final shouldDownloadOverMobileData =
Configuration.instance.shouldBackupOverMobileData();
_mlFramework = kCurrentModel == Model.onnxClip
_mlFramework = (kCurrentModel == Model.onnxClip && !kIsGraphene)
? ONNX(shouldDownloadOverMobileData)
: GGML(shouldDownloadOverMobileData);
await EmbeddingsDB.instance.init();