sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update CLIP model load path

Browse source
This commit is contained in:
Debanjum Singh Solanky 2021-10-02 16:50:06 -07:00
parent 31e21a6e4e
commit f59e321419
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/search_type/image_search.py
View file

@ -18,7 +18,7 @@ from src.utils.config import ImageSearchModel, ImageSearchConfig
def initialize_model():
# Initialize Model
torch.set_num_threads(4)
encoder = SentenceTransformer('clip-ViT-B-32') #Load the CLIP model
encoder = SentenceTransformer('sentence-transformers/clip-ViT-B-32') #Load the CLIP model
return encoder