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

Fix key error model_name issue by upgrade sentence-transformers version

Browse Source 
Refer to https://github.com/UKPLab/sentence-transformers/issues/1241
Also user verbose flag passed through function parameters in image_search
This commit is contained in:
Saba
2021-12-11 11:58:19 -05:00
parent ce7a751e6b
commit 9536358d34
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environment.yml
View File

@@ -6,7 +6,7 @@ dependencies:
- numpy=1.* - numpy=1.*
- pytorch=1.* - pytorch=1.*
- transformers=4.* - transformers=4.*
- sentence-transformers=2.0.0 - sentence-transformers=2.1.0
- fastapi=0.* - fastapi=0.*
- uvicorn=0.* - uvicorn=0.*
- pyyaml=5.* - pyyaml=5.*

2
src/search_type/image_search.py
View File

@@ -160,7 +160,7 @@ def setup(config: ImageSearchConfig, regenerate: bool, verbose: bool) -> ImageSe
# Extract Entries # Extract Entries
image_directory = resolve_absolute_path(config.input_directory, strict=True) image_directory = resolve_absolute_path(config.input_directory, strict=True)
image_names = extract_entries(image_directory, config.verbose) image_names = extract_entries(image_directory, verbose)
# Compute or Load Embeddings # Compute or Load Embeddings
embeddings_file = resolve_absolute_path(config.embeddings_file) embeddings_file = resolve_absolute_path(config.embeddings_file)