mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-10-18 07:29:45 +00:00
use CLIPImageProcessorPil to silence transformers torchvision warning (#24636)
This commit is contained in:
parent
5c32af3c7f
commit
186a730790
@ -5,7 +5,7 @@ import os
|
||||
import shutil
|
||||
import threading
|
||||
|
||||
from transformers import AutoTokenizer, CLIPImageProcessor
|
||||
from transformers import AutoTokenizer, CLIPImageProcessorPil
|
||||
from transformers.utils.logging import disable_progress_bar
|
||||
|
||||
from frigate.comms.inter_process import InterProcessRequestor
|
||||
@ -209,7 +209,7 @@ class JinaV1ImageEmbedding(BaseEmbedding):
|
||||
if self.downloader:
|
||||
self.downloader.wait_for_download()
|
||||
|
||||
self.feature_extractor = CLIPImageProcessor.from_pretrained(
|
||||
self.feature_extractor = CLIPImageProcessorPil.from_pretrained(
|
||||
f"{MODEL_CACHE_DIR}/{self.model_name}",
|
||||
)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user