klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-03 05:29:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Load model from HuggingFace if model_directory unset in config YAML

Browse Source 
- Do not save/load the model to/from disk when model_directory unset
in config.yml
- Add symmetric search default config to cli.py
This commit is contained in:
Debanjum Singh Solanky
2022-01-14 17:32:45 -05:00
parent 510faa1904
commit c64e0c2965
2 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/utils/cli.py
View File

@@ -77,14 +77,22 @@ default_config = {
},
'search-type':
{
'asymmetric':
'symmetric':
{
'encoder': "sentence-transformers/paraphrase-MiniLM-L6-v2",
'cross-encoder': "cross-encoder/ms-marco-MiniLM-L-6-v2",
'model_directory': None
},
'asymmetric':
{
'encoder': "sentence-transformers/msmarco-MiniLM-L-6-v3",
'cross-encoder': "cross-encoder/ms-marco-MiniLM-L-6-v2"
'cross-encoder': "cross-encoder/ms-marco-MiniLM-L-6-v2",
'model_directory': None
},
'image':
{
'encoder': "clip-ViT-B-32"
'encoder': "clip-ViT-B-32",
'model_directory': None
},
},
'processor':