fastembed-gpu 0.8.0__tar.gz → 0.9.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (76) hide show
  1. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/LICENSE +1 -1
  2. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/NOTICE +4 -2
  3. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/PKG-INFO +6 -5
  4. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/README.md +2 -2
  5. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/common/model_description.py +1 -0
  6. fastembed_gpu-0.9.0/fastembed/common/model_management.py +752 -0
  7. fastembed_gpu-0.9.0/fastembed/common/onnx_external_data.py +91 -0
  8. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/common/onnx_model.py +40 -4
  9. fastembed_gpu-0.9.0/fastembed/common/preprocessor_utils.py +159 -0
  10. fastembed_gpu-0.9.0/fastembed/common/utils.py +97 -0
  11. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/image_embedding.py +7 -1
  12. fastembed_gpu-0.9.0/fastembed/image/normalized_embedding.py +69 -0
  13. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/onnx_embedding.py +3 -3
  14. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/onnx_image_model.py +5 -1
  15. fastembed_gpu-0.9.0/fastembed/image/siglip_embedding.py +44 -0
  16. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/transform/functional.py +27 -10
  17. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/transform/operators.py +31 -5
  18. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/colbert.py +40 -15
  19. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/late_interaction_text_embedding.py +7 -1
  20. fastembed_gpu-0.9.0/fastembed/late_interaction/lateon.py +128 -0
  21. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/token_embeddings.py +1 -1
  22. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/colmodernvbert.py +51 -46
  23. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/colpali.py +3 -3
  24. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding.py +11 -2
  25. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/onnx_multimodal_model.py +15 -1
  26. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/parallel_processor.py +32 -3
  27. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/postprocess/muvera.py +23 -9
  28. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/custom_text_cross_encoder.py +32 -1
  29. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/onnx_text_cross_encoder.py +19 -1
  30. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/onnx_text_model.py +17 -4
  31. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/bm25.py +65 -14
  32. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/bm42.py +16 -8
  33. fastembed_gpu-0.9.0/fastembed/sparse/if_splade.py +240 -0
  34. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/minicoil.py +31 -11
  35. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/sparse_embedding_base.py +1 -1
  36. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/sparse_text_embedding.py +9 -2
  37. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/splade_pp.py +2 -1
  38. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/utils/vocab_resolver.py +10 -3
  39. fastembed_gpu-0.9.0/fastembed/text/builtin_sentence_embedding.py +84 -0
  40. fastembed_gpu-0.9.0/fastembed/text/custom_text_embedding.py +165 -0
  41. fastembed_gpu-0.9.0/fastembed/text/last_token_normalized_embedding.py +94 -0
  42. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/multitask_embedding.py +1 -1
  43. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/onnx_embedding.py +72 -13
  44. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/onnx_text_model.py +26 -6
  45. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/pooled_embedding.py +7 -7
  46. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/pooled_normalized_embedding.py +24 -8
  47. fastembed_gpu-0.9.0/fastembed/text/siglip_embedding.py +72 -0
  48. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/text_embedding.py +10 -16
  49. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/pyproject.toml +6 -6
  50. fastembed_gpu-0.8.0/fastembed/common/model_management.py +0 -475
  51. fastembed_gpu-0.8.0/fastembed/common/preprocessor_utils.py +0 -84
  52. fastembed_gpu-0.8.0/fastembed/common/utils.py +0 -69
  53. fastembed_gpu-0.8.0/fastembed/text/custom_text_embedding.py +0 -97
  54. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/__init__.py +0 -0
  55. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/common/__init__.py +0 -0
  56. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/common/types.py +0 -0
  57. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/embedding.py +0 -0
  58. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/__init__.py +0 -0
  59. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/image/image_embedding_base.py +0 -0
  60. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/__init__.py +0 -0
  61. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/jina_colbert.py +0 -0
  62. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction/late_interaction_embedding_base.py +0 -0
  63. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/__init__.py +0 -0
  64. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding_base.py +0 -0
  65. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/postprocess/__init__.py +0 -0
  66. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/py.typed +0 -0
  67. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/__init__.py +0 -0
  68. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/text_cross_encoder.py +0 -0
  69. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/rerank/cross_encoder/text_cross_encoder_base.py +0 -0
  70. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/__init__.py +0 -0
  71. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/utils/minicoil_encoder.py +0 -0
  72. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/utils/sparse_vectors_converter.py +0 -0
  73. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/sparse/utils/tokenizer.py +0 -0
  74. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/__init__.py +0 -0
  75. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/clip_embedding.py +0 -0
  76. {fastembed_gpu-0.8.0 → fastembed_gpu-0.9.0}/fastembed/text/text_embedding_base.py +0 -0
@@ -186,7 +186,7 @@
186
186
  same "printed page" as the copyright notice for easier
187
187
  identification within third-party archives.
188
188
 
189
- Copyright [yyyy] [name of copyright owner]
189
+ Copyright 2026 Qdrant Solutions GmbH
190
190
 
191
191
  Licensed under the Apache License, Version 2.0 (the "License");
192
192
  you may not use this file except in compliance with the License.
@@ -12,8 +12,10 @@ This distribution includes the following Jina AI models, each with its respectiv
12
12
 
13
13
  These models are developed by Jina (https://jina.ai/) and are subject to Jina AI's licensing terms.
14
14
 
15
- This distribution includes the following Google models, each with its respective license:
16
- - vidore/colpali-v1.3
15
+ This distribution includes the following Gemma-based models, each with its respective license:
16
+ - Qdrant/colpali-v1.3-fp16
17
+ - License: gemma
18
+ - google/embeddinggemma-300m
17
19
  - License: gemma
18
20
 
19
21
  Gemma is provided under and subject to the Gemma Terms of Use found at ai.google.dev/gemma/terms
@@ -1,21 +1,22 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: fastembed-gpu
3
- Version: 0.8.0
3
+ Version: 0.9.0
4
4
  Summary: Fast, light, accurate library built for retrieval embedding generation
5
- License: Apache License
5
+ License: Apache-2.0
6
6
  License-File: LICENSE
7
7
  License-File: NOTICE
8
8
  Keywords: vector,embedding,neural,search,qdrant,sentence-transformers
9
9
  Author: Qdrant Team
10
10
  Author-email: info@qdrant.tech
11
11
  Requires-Python: >=3.10.0
12
- Classifier: License :: Other/Proprietary License
12
+ Classifier: License :: OSI Approved :: Apache Software License
13
13
  Classifier: Programming Language :: Python :: 3
14
14
  Classifier: Programming Language :: Python :: 3.10
15
15
  Classifier: Programming Language :: Python :: 3.11
16
16
  Classifier: Programming Language :: Python :: 3.12
17
17
  Classifier: Programming Language :: Python :: 3.13
18
18
  Classifier: Programming Language :: Python :: 3.14
19
+ Classifier: Programming Language :: Python :: 3.15
19
20
  Requires-Dist: huggingface-hub (>=0.20,<2.0)
20
21
  Requires-Dist: loguru (>=0.7.2,<0.8.0)
21
22
  Requires-Dist: mmh3 (>=4.1.0,<6.0.0)
@@ -210,10 +211,10 @@ query = "What is Qdrant?"
210
211
  model = LateInteractionMultimodalEmbedding(model_name="Qdrant/colpali-v1.3-fp16")
211
212
  doc_images_embeddings = list(model.embed_image(doc_images))
212
213
  # shape (2, 1030, 128)
213
- # [array([[-0.03353882, -0.02090454, ..., -0.15576172, -0.07678223]], dtype=float32)]
214
+ # [array([[-0.03354, -0.0209, ..., -0.1558, -0.0768]], dtype=float16)]
214
215
  query_embedding = model.embed_text(query)
215
216
  # shape (1, 20, 128)
216
- # [array([[-0.00218201, 0.14758301, ..., -0.02207947, 0.16833496]], dtype=float32)]
217
+ # [array([[-0.002182, 0.1476, ..., -0.02208, 0.1683]], dtype=float16)]
217
218
  ```
218
219
 
219
220
  ### 🔄 Rerankers
@@ -169,10 +169,10 @@ query = "What is Qdrant?"
169
169
  model = LateInteractionMultimodalEmbedding(model_name="Qdrant/colpali-v1.3-fp16")
170
170
  doc_images_embeddings = list(model.embed_image(doc_images))
171
171
  # shape (2, 1030, 128)
172
- # [array([[-0.03353882, -0.02090454, ..., -0.15576172, -0.07678223]], dtype=float32)]
172
+ # [array([[-0.03354, -0.0209, ..., -0.1558, -0.0768]], dtype=float16)]
173
173
  query_embedding = model.embed_text(query)
174
174
  # shape (1, 20, 128)
175
- # [array([[-0.00218201, 0.14758301, ..., -0.02207947, 0.16833496]], dtype=float32)]
175
+ # [array([[-0.002182, 0.1476, ..., -0.02208, 0.1683]], dtype=float16)]
176
176
  ```
177
177
 
178
178
  ### 🔄 Rerankers
@@ -49,4 +49,5 @@ class SparseModelDescription(BaseModelDescription):
49
49
  class PoolingType(str, Enum):
50
50
  CLS = "CLS"
51
51
  MEAN = "MEAN"
52
+ LAST_TOKEN = "LAST_TOKEN"
52
53
  DISABLED = "DISABLED"