Ig nddlmZmZmZmZmZmZmZddlZ ddl m Z m Z ddl mZddlmZGddeZy))AnyDictIterableListOptionalSequenceTypeN) ImageInput OnnxProvider)ImageEmbeddingBase)OnnxImageEmbeddingceZdZUegZeeeed<e dee e e ffdZ dde dee deedeeeded eeed effd Z dd ed edeedeej.fdZxZS)ImageEmbeddingEMBEDDINGS_REGISTRYreturncjg}|jD]!}|j|j#|S)a Lists the supported models. Returns: List[Dict[str, Any]]: A list of dictionaries containing the model information. Example: ``` [ { "model": "Qdrant/clip-ViT-B-32-vision", "dim": 512, "description": "CLIP vision encoder based on ViT-B/32", "license": "mit", "size_in_GB": 0.33, "sources": { "hf": "Qdrant/clip-ViT-B-32-vision", }, "model_file": "model.onnx", } ] ``` )rextendlist_supported_models)clsresult embeddings \/var/www/html/answerous/venv/lib/python3.12/site-packages/fastembed/image/image_embedding.pyrz$ImageEmbedding.list_supported_models s8200 =I MM)99; < = model_name cache_dirthreads providerscuda device_ids lazy_loadc t |||fi||jD]=} | j} t fd| Ds(| |f|||||d||_yt dd)Nc3fK|](}j|djk(*yw)modelN)lower).0r#rs r z*ImageEmbedding.__init__..9s,^E:##%w)=)=)??^s.1)rrrrr zModel zt is not supported in ImageEmbedding.Please check the supported models using `ImageEmbedding.list_supported_models()`)super__init__rranyr# ValueError) selfrrrrrrr kwargsEMBEDDING_MODEL_TYPEsupported_models __class__s ` rr(zImageEmbedding.__init__+s YB6B$($<$<  3IIK ^M]^^1 $')'    ZL!_ _  rimages batch_sizeparallelc+^K|jj|||fi|Ed{y7w)a Encode a list of documents into list of embeddings. We use mean pooling with attention so that the model can handle variable-length inputs. Args: images: Iterator of image paths or single image path to embed batch_size: Batch size for encoding -- higher values will use more memory, but be faster parallel: If > 1, data-parallel encoding will be used, recommended for offline encoding of large datasets. If 0, use all available cores. If None, don't use data-parallel processing, use default onnxruntime threading instead. Returns: List of embeddings, one per document N)r#embed)r+r0r1r2r,s rr4zImageEmbedding.embedKs+,$4::##FJKFKKKs #-+-)NNNFNF)N)__name__ __module__ __qualname__r rrr r __annotations__ classmethodrstrrrrintrr boolr(r rnpndarrayr4 __classcell__)r/s@rrr s;M:Nd#567Nd4S>&:@$(!%6:*.  C= #  H\23    T#Y'  F"& LLL3- L "**  Lrr)typingrrrrrrr numpyr>fastembed.commonr r $fastembed.image.image_embedding_baser fastembed.image.onnx_embeddingr rrrrGs+FFF5C=WL'WLr