corvic-engine 0.3.0rc83__tar.gz → 0.3.0rc84__tar.gz

{corvic_engine-0.3.0rc83 → corvic_engine-0.3.0rc84}/Cargo.lock

@@ -16,7 +16,7 @@ checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
 [[package]]
 name = "corvic-engine"
-version = "0.3.0-rc.83"
+version = "0.3.0-rc.84"
 dependencies = [
  "numpy",
  "pyo3",
@@ -57,7 +57,7 @@ checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
 
 [[package]]
 name = "engine-walk"
-version = "0.3.0-rc.83"
+version = "0.3.0-rc.84"
 dependencies = [
  "version_check",
 ]
@@ -165,9 +165,9 @@ checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
 
 [[package]]
 name = "portable-atomic"
-version = "1.11.0"
+version = "1.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "350e9b48cbc6b0e028b0473b114454c6316e57336ee184ceab6e53f72c178b3e"
+checksum = "f84267b20a16ea918e43c6a88433c2d54fa145c92a811b5b047ccbe153674483"
 
 [[package]]
 name = "portable-atomic-util"

{corvic_engine-0.3.0rc83 → corvic_engine-0.3.0rc84}/Cargo.toml

@@ -3,7 +3,7 @@ resolver = "2"
 members = ["python"]
 
 [workspace.package]
-version = "0.3.0-rc.83"
+version = "0.3.0-rc.84"
 edition = "2021"
 publish = false
 

{corvic_engine-0.3.0rc83 → corvic_engine-0.3.0rc84}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corvic-engine
-Version: 0.3.0rc83
+Version: 0.3.0rc84
 Classifier: Environment :: Console
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: Implementation :: CPython
@@ -27,6 +27,7 @@ Requires-Dist: pillow>=10.0.0 ; extra == 'ml'
 Requires-Dist: scikit-learn>=1.4.0 ; extra == 'ml'
 Requires-Dist: transformers[torch]>=4.45.0 ; extra == 'ml'
 Requires-Dist: sentencepiece>=0.2.0 ; extra == 'ml'
+Requires-Dist: torchvision==0.22.1 ; extra == 'ml'
 Requires-Dist: opentelemetry-api>=1.20.0 ; extra == 'telemetry'
 Requires-Dist: opentelemetry-sdk>=1.20.0 ; extra == 'telemetry'
 Provides-Extra: ml

{corvic_engine-0.3.0rc83 → corvic_engine-0.3.0rc84}/pyproject.toml

@@ -18,7 +18,7 @@ authors = [
 ]
 readme = "README.md"
 description = "Seamless embedding generation and retrieval."
-version = "0.3.0-rc.83" # sync this with tool.poetry.version below
+version = "0.3.0-rc.84" # sync this with tool.poetry.version below
 
 # TODO(ddn): Pull dependencies from tool.poetry.dependencies. We use maturin as
 # the build system but poetry for the development environment.
@@ -52,6 +52,7 @@ ml = [
   "scikit-learn >= 1.4.0",
   "transformers[torch] >= 4.45.0", # Needed for pre-trained torch models
   "sentencepiece >= 0.2.0", # Needed for SigLIP2
+  "torchvision == 0.22.1",
 ]
 
 telemetry = [
@@ -72,7 +73,7 @@ packages = [
 ]
 description = ""
 authors = []
-version = "0.3.0-rc.83"
+version = "0.3.0-rc.84"
 
 [tool.poetry.scripts]
 check = "corvic_check.cli:main"
@@ -157,6 +158,7 @@ scikit-learn = "^1.4.0"
 optional = true
 
 [tool.poetry.group.ml.dependencies]
+sentencepiece = "^0.2.0"
 torch = [
   { platform = "linux", version = "^2.7.1", source = "torch-cpu-linux" },
   { platform = "darwin", version = "^2.7.1", source = "pypi" },

{corvic_engine-0.3.0rc83 → corvic_engine-0.3.0rc84}/python/corvic/system/_image_embedder.py

@@ -182,6 +182,8 @@ class Clip(HFModelImageEmbedder):
         return "5812e510083bb2d23fa43778a39ac065d205ed4d"
 
     def _load_models(self) -> LoadedModels:
+        from transformers.models.auto.modeling_auto import AutoModel
+        from transformers.models.auto.processing_auto import AutoProcessor
         from transformers.models.clip import (
             CLIPModel,
             CLIPProcessor,