^Nj UddlmZmZmZddlmZddlmZddlm Z m Z ddl m Z m Z e dddd d e d d gZee ed<Gdde ZGdde Zy))AnyIterableType) NumpyArray)OnnxOutputContext)OnnxTextEmbeddingOnnxTextEmbeddingWorker)DenseModelDescription ModelSourcezQdrant/clip-ViT-B-32-textizText embeddings, Multimodal (text&image), English, 77 input tokens truncation, Prefixes for queries/documents: not necessary, 2021 yearmitg?)hfz model.onnx)modeldim descriptionlicense size_in_GBsources model_filesupported_clip_modelsc^eZdZedeefdZedeefdZ de de de e fdZy)CLIPOnnxEmbeddingreturnctSN)CLIPEmbeddingWorkerclss n/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastembed/text/clip_embedding.py_get_worker_classz#CLIPOnnxEmbedding._get_worker_classs""ctS)zLists the supported models. Returns: list[DenseModelDescription]: A list of DenseModelDescription objects containing the model information. )rrs r_list_supported_modelsz(CLIPOnnxEmbedding._list_supported_modelss %$r outputkwargsc |jSr) model_output)selfr#r$s r_post_process_onnx_outputz+CLIPOnnxEmbedding._post_process_onnx_output&s"""r N)__name__ __module__ __qualname__ classmethodrr rlistr r"rrrrr(r rrrsd#$'>"?##%t,A'B%%#'#36# * #r rc$eZdZdedededefdZy)r model_name cache_dirr$rc td||dd|S)N)r0r1threadsr.)r)r'r0r1r$s rinit_embeddingz"CLIPEmbeddingWorker.init_embedding-s) ! !    r N)r)r*r+strrrr5r.r rrr,s-          r rN)typingrrrfastembed.common.typesrfastembed.common.onnx_modelrfastembed.text.onnx_embeddingrr "fastembed.common.model_descriptionr r rr-__annotations__rrr.r rr=sp&&-9TQ)  G:;  6t12  #)#(  1  r