sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 15:38:55 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
environment.yml
View file

@ -6,7 +6,7 @@ dependencies:
- numpy=1.*
- pytorch=1.*
- transformers=4.*
- sentence-transformers=2.0.0
- sentence-transformers=2.1.0
- fastapi=0.*
- uvicorn=0.*
- 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
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
embeddings_file = resolve_absolute_path(config.embeddings_file)