sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

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.* - 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)