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

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
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(): def initialize_model():
# Initialize Model # Initialize Model
torch.set_num_threads(4) 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 return encoder