Fix access (#14540)

This commit is contained in:
Nicolas Mowen 2024-10-23 09:26:03 -06:00
parent 5be98a2457
commit 901ba8ece6

View File

@ -116,10 +116,10 @@ class Embeddings:
model_name="jinaai/jina-clip-v1", model_name="jinaai/jina-clip-v1",
model_file=model_file, model_file=model_file,
download_urls=download_urls, download_urls=download_urls,
model_size=config.model_size, model_size=config.semantic_search.model_size,
model_type=ModelTypeEnum.vision, model_type=ModelTypeEnum.vision,
requestor=self.requestor, requestor=self.requestor,
device="GPU" if config.model_size == "large" else "CPU", device="GPU" if config.semantic_search.model_size == "large" else "CPU",
) )
self.face_embedding = None self.face_embedding = None