Upload folder using huggingface_hub

README.md

@@ -38,7 +38,8 @@ base_model: Scentience-OVL-Embeddings-Base
 
 ## Description
 
-This repository is the first-ever series of multimodal joint embedding models trained on olfaction, vision, and language data specifically for tasks in robotics and embodied artificial intelligence.
+This repository is a foundational series of multimodal joint embedding models trained on olfaction, vision, and language data.
+These models are built specifically for prototyping and exploratory tasks within AR/VR, robotics, and embodied artificial intelligence.
 
 
 To the best of our knowledge, there are currently no open-source datasets that provide aligned olfactory, visual, and linguistic annotations, making this the first.
@@ -55,9 +56,9 @@ Whether these models are used for better vision-scent navigation with drones, tr
 ## Models
 We offer four olfaction-vision-language (OVL) embedding models with this repository:
  - (1) `ovle-large-base`: The original OVL base model. This model is optimal for online tasks where accuracy is critical.
- - (2) `ovle-large-graph`: The OVL base model built around a graph-attention-convolution network. This model is optimal for online tasks where accuracy is paramount and inference time is not as critical.
+ - (2) `ovle-large-gat`: The OVL base model built around a graph-attention-convolution network. This model is optimal for online tasks where accuracy is paramount and inference time is not as critical.
  - (3) `ovle-small-base`: The original OVL base model optimized for faster inference and edge-based robotics. This model is optimized for export to common frameworks that run on Android, iOS, Rust, and others.
- - (4) `ovle-small-graph`: The OVL graph-attention-convolution model optimized for faster inference and edge robotics applications.
+ - (4) `ovle-small-gat`: The OVL graph-attention-convolution model optimized for faster inference and edge robotics applications.
 
 ## Training Data
 A sample dataset is included, but the full datasets are linked in the `Datasets` pane of this repo.

model/ovle-large/base/gnn_2048.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f09870edab48b3a268f67cd59d6fcb0999b5e564d58349b5d96a53a10ba4778
+size 117501257

model/ovle-large/base/gnn_2048.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b515ad6d6ff83bdd3bf21729bb9f9a2895557d4e7decab60dc82b5f4b5c18c3
+size 100713428

model/ovle-large/base/olf_encoder_2048.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a01392bbfe4cdc14de5d39c4a32fc429de56dd98d5881d48846771711e88d87
+size 2138537

model/ovle-large/base/olf_encoder_2048.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a563630e0ed987db639dfd450ee4c4831ea2a15844ed9d8d7f6bd798e99b6987
+size 2123960

model/ovle-large/gat/gnn_gat_2048.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c1f4b5f55565b53bf68482c42ebde9221c9a933e8f9d45dc1c894fc55102321
+size 570948421

model/ovle-large/gat/gnn_gat_2048.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:284de5244a2c83a51c82f1a15ea2324cdc7e34fdc7006227f0a7b2fc7fce635c
+size 537297732

model/ovle-large/gat/olf_encoder_gat_2048.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce87939ab0c5ca4b8394b33149d68ba0d176f045030ca8504db0e35534a7a0d2
+size 2138889

model/ovle-large/gat/olf_encoder_gat_2048.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c354b23e7e4181308800a1afd35789b91f1ec31b08eb5dd95565ae2a5bd0d442
+size 2123960

model/ovle-small/base/gnn_512.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:495f83e249b82415e45d4a1157c23a8eb7d6037247ac90ad407916d65ef3cbf7
+size 7363884

model/ovle-small/base/gnn_512.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b75424968155035ae6ba2207a351dcdb7a7de396901c9321c6eef8e38e831f09
+size 6304684

model/ovle-small/base/olf_encoder_512.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcbb51a6ef79cb1ce3c7c7a666baba9daa965ac464f9365ae94c41ce820b4669
+size 1284474

model/ovle-small/base/olf_encoder_512.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ccfbdbc16a72894a889eff76c45929483dc10ed68c29eed0812abcf9fe8bdcd
+size 1269944

model/ovle-small/gat/gnn_gat_512.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14cf669fe71b766e20620353d015d7eee35931f762508af74c1cf3848ef14a98
+size 35848956

model/ovle-small/gat/gnn_gat_512.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3336b820a021d0f1fccd34022431307779f39b3ddefc6decf30e706b9e457fd5
+size 33661740

model/ovle-small/gat/olf_encoder_gat_512.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9b611b0b92403200bd4cb65c3cff483cd7c7c07b6b483ecb4f3f0a716dda2f67
+size 1284826

model/ovle-small/gat/olf_encoder_gat_512.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5da86d1ac27af8eeb01096a6dfbd04459dbe65b2ae84cf6670f413b38ff1f2c5
+size 1269944

model_cards/ovle-large.md

@@ -10,7 +10,8 @@
   - **Vision encoder:** CLIP-based
   - **Language encoder:** CLIP-based
   - **Fusion strategy:** Joint embedding space via multimodal contrastive training
-  - **Parameter Count:** 134.9M (without CLIP), 286.1M (with CLIP)
+  - **Parameter Count (Base):** 29.9M (without CLIP), 181.2M (with CLIP)
+  - **Parameter Count (GAT):** 143.2M (without CLIP), 294.5M (with CLIP)
   - **Embedding Dimension:** 2048
 - **License:** MIT
 - **Contact:** [email protected], [email protected]

model_cards/ovle-small.md

@@ -10,7 +10,8 @@
   - **Vision encoder:** CLIP-based
   - **Language encoder:** CLIP-based
   - **Fusion strategy:** Joint embedding space via multimodal contrastive training
-  - **Parameter Count:** 8.7M (without CLIP), 160.0M (with CLIP)
+  - **Parameter Count (Base):** 2.2M (without CLIP), 153.4M (with CLIP)
+  - **Parameter Count (GAT):** 9.3M (without CLIP), 160.5M (with CLIP)
   - **Embedding Dimension:** 512
 - **License:** MIT
 - **Contact:** [email protected], [email protected]

notebooks/olfaction_vision_language_embeddings_inference.ipynb

@@ -43,7 +43,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 1,
+      "execution_count": null,
       "metadata": {
         "id": "cxabQxw9LSzM",
         "colab": {
@@ -161,13 +161,13 @@
         "DEVICE = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
         "BATCH_SIZE = 16\n",
         "EMBED_DIM = 512     # Embedding dims = 512 for small, 2048 for large\n",
-        "ENCODER_FILE_PATH = \"./olf_encoder.safetensors\"\n",
-        "GNN_FILE_PATH = \"./gnn.safetensors\""
+        "ENCODER_FILE_PATH = \"./olf_encoder.pt\"\n",
+        "GNN_FILE_PATH = \"./gnn.pt\""
       ],
       "metadata": {
         "id": "BHUd6n3bLbqo"
       },
-      "execution_count": 2,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -219,7 +219,7 @@
       "metadata": {
         "id": "UzhGG8CzMmBs"
       },
-      "execution_count": 3,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -235,8 +235,8 @@
       "cell_type": "code",
       "source": [
         "# Load the models\n",
-        "olf_encoder = load_file(ENCODER_FILE_PATH)\n",
-        "graph_model = load_file(GNN_FILE_PATH)\n",
+        "olf_encoder = torch.jit.load(ENCODER_FILE_PATH)\n",
+        "graph_model = torch.jit.load(GNN_FILE_PATH)\n",
         "clip_model = CLIPModel.from_pretrained(\"openai/clip-vit-base-patch32\").to(DEVICE)\n",
         "\n",
         "# Build example vision-olfaction sample with dummy data\n",
@@ -257,7 +257,7 @@
       "metadata": {
         "id": "_U5qqxn8Mibo"
       },
-      "execution_count": 4,
+      "execution_count": null,
       "outputs": []
     }
   ]

requirements.txt

@@ -1,608 +1,24 @@
-absl-py==1.4.0
-absolufy-imports==0.3.1
-accelerate==1.10.1
-aiofiles==24.1.0
-aiohappyeyeballs==2.6.1
-aiohttp==3.12.15
-aiosignal==1.4.0
-alabaster==1.0.0
-albucore==0.0.24
-albumentations==2.0.8
-ale-py==0.11.2
-alembic==1.16.5
-altair==5.5.0
-annotated-types==0.7.0
-antlr4-python3-runtime==4.9.3
-anyio==4.10.0
-anywidget==0.9.18
-argon2-cffi==25.1.0
-argon2-cffi-bindings==25.1.0
-array_record==0.8.1
-arrow==1.3.0
-arviz==0.22.0
-astropy==7.1.0
-astropy-iers-data==0.2025.9.1.0.42.11
-astunparse==1.6.3
-atpublic==5.1
-attrs==25.3.0
-audioread==3.0.1
-Authlib==1.6.3
-autograd==1.8.0
-babel==2.17.0
-backcall==0.2.0
-beartype==0.21.0
-beautifulsoup4==4.13.5
-betterproto==2.0.0b6
-bigframes==2.18.0
-bigquery-magics==0.10.3
-bleach==6.2.0
-blinker==1.9.0
-blis==1.3.0
-blobfile==3.0.0
-blosc2==3.7.2
-bokeh==3.7.3
-Bottleneck==1.4.2
-bqplot==0.12.45
-branca==0.8.1
-Brotli==1.1.0
-build==1.3.0
-CacheControl==0.14.3
-cachetools==5.5.2
-catalogue==2.0.10
-certifi==2025.8.3
-cffi==1.17.1
-chardet==5.2.0
-charset-normalizer==3.4.3
-chex==0.1.90
-clarabel==0.11.1
-click==8.2.1
-cloudpathlib==0.22.0
-cloudpickle==3.1.1
-cmake==3.31.6
-cmdstanpy==1.2.5
-colorcet==3.1.0
-colorlover==0.3.0
-colour==0.1.5
-community==1.0.0b1
-confection==0.1.5
-cons==0.4.7
-contourpy==1.3.3
-cramjam==2.11.0
-cryptography==43.0.3
-cuda-python==12.6.2.post1
-cudf-cu12 @ https://pypi.nvidia.com/cudf-cu12/cudf_cu12-25.6.0-cp312-cp312-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
-cudf-polars-cu12==25.6.0
-cufflinks==0.17.3
-cuml-cu12==25.6.0
-cupy-cuda12x==13.3.0
-curl_cffi==0.13.0
-cuvs-cu12==25.6.1
-cvxopt==1.3.2
-cvxpy==1.6.7
-cycler==0.12.1
-cyipopt==1.5.0
-cymem==2.0.11
-Cython==3.0.12
-dask==2025.5.0
-dask-cuda==25.6.0
-dask-cudf-cu12==25.6.0
-dataproc-spark-connect==0.8.3
-datasets==4.0.0
-db-dtypes==1.4.3
-dbus-python==1.2.18
-debugpy==1.8.15
-decorator==4.4.2
-defusedxml==0.7.1
-diffusers==0.35.1
-dill==0.3.8
-distributed==2025.5.0
-distributed-ucxx-cu12==0.44.0
-distro==1.9.0
-dlib==19.24.6
-dm-tree==0.1.9
-docstring_parser==0.17.0
-docutils==0.21.2
-dopamine_rl==4.1.2
-duckdb==1.3.2
-earthengine-api==1.5.24
-easydict==1.13
-editdistance==0.8.1
-eerepr==0.1.2
-einops==0.8.1
-en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0-py3-none-any.whl#sha256=1932429db727d4bff3deed6b34cfc05df17794f4a52eeb26cf8928f7c1a0fb85
-entrypoints==0.4
-et_xmlfile==2.0.0
-etils==1.13.0
-etuples==0.3.10
-Farama-Notifications==0.0.4
-fastai==2.8.4
-fastapi==0.116.1
-fastcore==1.8.8
-fastdownload==0.0.7
-fastjsonschema==2.21.2
-fastprogress==1.0.3
-fastrlock==0.8.3
-fasttransform==0.0.2
-ffmpy==0.6.1
-filelock==3.19.1
-firebase-admin==6.9.0
-Flask==3.1.2
-flatbuffers==25.2.10
-flax==0.10.6
-folium==0.20.0
-fonttools==4.59.2
-fqdn==1.5.1
-frozendict==2.4.6
-frozenlist==1.7.0
-fsspec==2025.3.0
-future==1.0.0
-gast==0.6.0
-gcsfs==2025.3.0
-GDAL==3.8.4
-gdown==5.2.0
-geemap==0.35.3
-geocoder==1.38.1
-geographiclib==2.1
-geopandas==1.1.1
-geopy==2.4.1
-gin-config==0.5.0
-gitdb==4.0.12
-GitPython==3.1.45
-glob2==0.7
-google==2.0.3
-google-adk==1.13.0
-google-ai-generativelanguage==0.6.15
-google-api-core==2.25.1
-google-api-python-client==2.181.0
-google-auth==2.38.0
-google-auth-httplib2==0.2.0
-google-auth-oauthlib==1.2.2
-google-cloud-aiplatform==1.111.0
-google-cloud-appengine-logging==1.6.2
-google-cloud-audit-log==0.3.2
-google-cloud-bigquery==3.36.0
-google-cloud-bigquery-connection==1.18.3
-google-cloud-bigquery-storage==2.33.0
-google-cloud-bigtable==2.32.0
-google-cloud-core==2.4.3
-google-cloud-dataproc==5.21.0
-google-cloud-datastore==2.21.0
-google-cloud-firestore==2.21.0
-google-cloud-functions==1.20.4
-google-cloud-language==2.17.2
-google-cloud-logging==3.12.1
-google-cloud-resource-manager==1.14.2
-google-cloud-secret-manager==2.24.0
-google-cloud-spanner==3.57.0
-google-cloud-speech==2.33.0
-google-cloud-storage==2.19.0
-google-cloud-trace==1.16.2
-google-cloud-translate==3.21.1
-google-colab @ file:///colabtools/dist/google_colab-1.0.0.tar.gz
-google-crc32c==1.7.1
-google-genai==1.33.0
-google-generativeai==0.8.5
-google-pasta==0.2.0
-google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
-googledrivedownloader==1.1.0
-gradio==5.44.1
-gradio_client==1.12.1
-graphviz==0.21
-greenlet==3.2.4
-groovy==0.1.2
-grpc-google-iam-v1==0.14.2
-grpc-interceptor==0.15.4
-grpcio==1.74.0
-grpcio-status==1.71.2
-grpclib==0.4.8
-gspread==6.2.1
-gspread-dataframe==4.0.0
-gym==0.25.2
-gym-notices==0.1.0
-gymnasium==1.2.0
-h11==0.16.0
-h2==4.3.0
-h5netcdf==1.6.4
-h5py==3.14.0
-hdbscan==0.8.40
 hf-xet==1.1.9
 hf_transfer==0.1.9
-highspy==1.11.0
-holidays==0.80
-holoviews==1.21.0
-hpack==4.1.0
-html5lib==1.1
-httpcore==1.0.9
-httpimport==1.4.1
-httplib2==0.30.0
-httpx==0.28.1
-httpx-sse==0.4.1
 huggingface-hub==0.34.4
-humanize==4.13.0
-hyperframe==6.1.0
-hyperopt==0.2.7
-ibis-framework==9.5.0
-idna==3.10
-imageio==2.37.0
-imageio-ffmpeg==0.6.0
-imagesize==1.4.1
-imbalanced-learn==0.14.0
-immutabledict==4.2.1
-importlib_metadata==8.7.0
-importlib_resources==6.5.2
-imutils==0.5.4
-inflect==7.5.0
-iniconfig==2.1.0
-intel-cmplr-lib-ur==2025.2.1
-intel-openmp==2025.2.1
-ipyevents==2.0.2
-ipyfilechooser==0.6.0
-ipykernel==6.17.1
-ipyleaflet==0.20.0
-ipyparallel==8.8.0
-ipython==7.34.0
-ipython-genutils==0.2.0
-ipython-sql==0.5.0
-ipytree==0.2.2
-ipywidgets==7.7.1
-isoduration==20.11.0
-itsdangerous==2.2.0
-jaraco.classes==3.4.0
-jaraco.context==6.0.1
-jaraco.functools==4.3.0
-jax==0.5.3
-jax-cuda12-pjrt==0.5.3
-jax-cuda12-plugin==0.5.3
-jaxlib==0.5.3
-jeepney==0.9.0
-jieba==0.42.1
-Jinja2==3.1.6
-jiter==0.10.0
-joblib==1.5.2
-jsonpatch==1.33
-jsonpickle==4.1.1
-jsonpointer==3.0.0
-jsonschema==4.25.1
-jsonschema-specifications==2025.4.1
-jupyter-console==6.6.3
-jupyter-events==0.12.0
-jupyter-leaflet==0.20.0
-jupyter_client==7.4.9
-jupyter_core==5.8.1
-jupyter_kernel_gateway @ git+https://github.com/googlecolab/kernel_gateway@b134e9945df25c2dcb98ade9129399be10788671
-jupyter_server==2.14.0
-jupyter_server_terminals==0.5.3
-jupyterlab_pygments==0.3.0
-jupyterlab_widgets==3.0.15
-jupytext==1.17.3
-kaggle==1.7.4.5
-kagglehub==0.3.13
-keras==3.10.0
-keras-hub==0.21.1
-keras-nlp==0.21.1
-keyring==25.6.0
-keyrings.google-artifactregistry-auth==1.1.2
-kiwisolver==1.4.9
-langchain==0.3.27
-langchain-core==0.3.75
-langchain-text-splitters==0.3.11
-langcodes==3.5.0
-langsmith==0.4.24
-language_data==1.3.0
-lark==1.2.2
-launchpadlib==1.10.16
-lazr.restfulclient==0.14.4
-lazr.uri==1.0.6
-lazy_loader==0.4
-libclang==18.1.1
-libcudf-cu12 @ https://pypi.nvidia.com/libcudf-cu12/libcudf_cu12-25.6.0-py3-none-manylinux_2_28_x86_64.whl
-libcugraph-cu12==25.6.0
-libcuml-cu12==25.6.0
-libcuvs-cu12==25.6.1
-libkvikio-cu12==25.6.0
-libpysal==4.13.0
-libraft-cu12==25.6.0
-librmm-cu12==25.6.0
-librosa==0.11.0
-libucx-cu12==1.18.1
-libucxx-cu12==0.44.0
-lightgbm @ file:///tmp/lightgbm/LightGBM/dist/lightgbm-4.6.0-py3-none-linux_x86_64.whl
-linkify-it-py==2.0.3
-llvmlite==0.43.0
-locket==1.0.0
-logical-unification==0.4.6
-lxml==5.4.0
-Mako==1.3.10
-marisa-trie==1.3.1
-Markdown==3.9
-markdown-it-py==4.0.0
-MarkupSafe==3.0.2
 matplotlib==3.10.0
-matplotlib-inline==0.1.7
-matplotlib-venn==1.1.2
-mcp==1.13.1
-mdit-py-plugins==0.5.0
-mdurl==0.1.2
-miniKanren==1.0.5
-missingno==0.5.2
-mistune==3.1.4
-mizani==0.13.5
-mkl==2025.2.0
-ml_dtypes==0.5.3
-mlxtend==0.23.4
-more-itertools==10.8.0
-moviepy==1.0.3
-mpmath==1.3.0
-msgpack==1.1.1
-multidict==6.6.4
-multipledispatch==1.0.0
-multiprocess==0.70.16
-multitasking==0.0.12
-murmurhash==1.0.13
-music21==9.3.0
-namex==0.1.0
-narwhals==2.3.0
-natsort==8.4.0
-nbclassic==1.3.2
-nbclient==0.10.2
-nbconvert==7.16.6
-nbformat==5.10.4
-ndindex==1.10.0
-nest-asyncio==1.6.0
-networkx==3.5
-nibabel==5.3.2
-nltk==3.9.1
-notebook==6.5.7
-notebook_shim==0.2.4
-numba==0.60.0
-numba-cuda==0.11.0
-numexpr==2.11.0
 numpy==2.0.2
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvcc-cu12==12.5.82
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.10.2.21
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.7.1
-nvidia-ml-py==12.575.51
-nvidia-nccl-cu12==2.27.3
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
-nvtx==0.2.13
-nx-cugraph-cu12 @ https://pypi.nvidia.com/nx-cugraph-cu12/nx_cugraph_cu12-25.6.0-py3-none-any.whl
-oauth2client==4.1.3
-oauthlib==3.3.1
-omegaconf==2.3.0
 openai==1.106.1
 opencv-contrib-python==4.12.0.88
 opencv-python==4.12.0.88
 opencv-python-headless==4.12.0.88
 openpyxl==3.1.5
-opentelemetry-api==1.36.0
-opentelemetry-exporter-gcp-trace==1.9.0
-opentelemetry-resourcedetector-gcp==1.9.0a0
-opentelemetry-sdk==1.36.0
-opentelemetry-semantic-conventions==0.57b0
 opt_einsum==3.4.0
-optax==0.2.5
-optree==0.17.0
-orbax-checkpoint==0.11.24
-orjson==3.11.3
-osqp==1.0.4
-overrides==7.7.0
-packaging==25.0
 pandas==2.2.2
 pandas-datareader==0.10.0
-pandas-gbq==0.29.2
-pandas-stubs==2.2.2.240909
-pandocfilters==1.5.1
-panel==1.7.5
-param==2.2.1
-parso==0.8.5
-parsy==2.1
-partd==1.4.2
-patsy==1.0.1
-peewee==3.18.2
-peft==0.17.1
-pexpect==4.9.0
-pickleshare==0.7.5
-pillow==11.3.0
-platformdirs==4.4.0
-plotly==5.24.1
-plotnine==0.14.5
-pluggy==1.6.0
-plum-dispatch==2.5.7
-ply==3.11
-polars==1.25.2
-pooch==1.8.2
-portpicker==1.5.2
-preshed==3.0.10
-prettytable==3.16.0
-proglog==0.1.12
-progressbar2==4.5.0
-prometheus_client==0.22.1
-promise==2.3
-prompt_toolkit==3.0.52
-propcache==0.3.2
-prophet==1.1.7
-proto-plus==1.26.1
-protobuf==5.29.5
-psutil==5.9.5
-psycopg2==2.9.10
-psygnal==0.14.1
-ptyprocess==0.7.0
-py-cpuinfo==9.0.0
-py4j==0.10.9.7
-pyarrow==18.1.0
-pyasn1==0.6.1
-pyasn1_modules==0.4.2
-pycairo==1.28.0
-pycocotools==2.0.10
-pycparser==2.22
-pycryptodomex==3.23.0
-pydantic==2.11.7
-pydantic-settings==2.10.1
-pydantic_core==2.33.2
-pydata-google-auth==1.9.1
-pydot==3.0.4
-pydotplus==2.0.2
-PyDrive2==1.21.3
-pydub==0.25.1
-pyerfa==2.0.1.5
-pygame==2.6.1
-pygit2==1.18.2
-Pygments==2.19.2
-PyGObject==3.42.0
-PyJWT==2.10.1
-pylibcudf-cu12 @ https://pypi.nvidia.com/pylibcudf-cu12/pylibcudf_cu12-25.6.0-cp312-cp312-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
-pylibcugraph-cu12==25.6.0
-pylibraft-cu12==25.6.0
-pymc==5.25.1
-pynndescent==0.5.13
-pynvjitlink-cu12==0.7.0
-pynvml==12.0.0
-pyogrio==0.11.1
-pyomo==6.9.4
-PyOpenGL==3.1.10
-pyOpenSSL==24.2.1
-pyparsing==3.2.3
-pyperclip==1.9.0
-pyproj==3.7.2
-pyproject_hooks==1.2.0
-pyshp==2.3.1
-PySocks==1.7.1
-pyspark==3.5.1
-pytensor==2.31.7
-pytest==8.4.2
-python-apt==0.0.0
-python-box==7.3.2
-python-dateutil==2.9.0.post0
-python-dotenv==1.1.1
-python-json-logger==3.3.0
-python-louvain==0.16
-python-multipart==0.0.20
-python-slugify==8.0.4
-python-snappy==0.7.3
-python-utils==3.9.1
-pytz==2025.2
-pyviz_comms==3.0.6
-PyWavelets==1.9.0
-PyYAML==6.0.2
-pyzmq==26.2.1
-raft-dask-cu12==25.6.0
-rapids-dask-dependency==25.6.0
-rapids-logger==0.1.1
-ratelim==0.1.6
-referencing==0.36.2
-regex==2024.11.6
 requests==2.32.4
-requests-oauthlib==2.0.0
-requests-toolbelt==1.0.0
-requirements-parser==0.9.0
-rfc3339-validator==0.1.4
-rfc3986-validator==0.1.1
-rfc3987-syntax==1.1.0
-rich==13.9.4
-rmm-cu12==25.6.0
-roman-numerals-py==3.1.0
-rpds-py==0.27.1
-rpy2==3.5.17
-rsa==4.9.1
-ruff==0.12.12
-safehttpx==0.1.6
 safetensors==0.6.2
 scikit-image==0.25.2
 scikit-learn==1.6.1
 scipy==1.16.1
-scooby==0.10.1
-scs==3.2.8
-seaborn==0.13.2
-SecretStorage==3.3.3
-semantic-version==2.10.0
-Send2Trash==1.8.3
-sentence-transformers==5.1.0
-sentencepiece==0.2.1
-sentry-sdk==2.36.0
-setuptools==75.2.0
-shap==0.48.0
-shapely==2.1.1
-shellingham==1.5.4
-simple-parsing==0.1.7
-simplejson==3.20.1
-simsimd==6.5.1
-six==1.17.0
 sklearn-pandas==2.2.0
-slicer==0.0.8
-smart_open==7.3.0.post1
-smmap==5.0.2
-sniffio==1.3.1
-snowballstemmer==3.0.1
-sortedcontainers==2.4.0
-soundfile==0.13.1
-soupsieve==2.8
-soxr==0.5.0.post1
-spacy==3.8.7
-spacy-legacy==3.0.12
-spacy-loggers==1.0.5
-spanner-graph-notebook==1.1.8
-Sphinx==8.2.3
-sphinxcontrib-applehelp==2.0.0
-sphinxcontrib-devhelp==2.0.0
-sphinxcontrib-htmlhelp==2.1.0
-sphinxcontrib-jsmath==1.0.1
-sphinxcontrib-qthelp==2.0.0
-sphinxcontrib-serializinghtml==2.0.0
-SQLAlchemy==2.0.43
-sqlalchemy-spanner==1.16.0
-sqlglot==25.20.2
-sqlparse==0.5.3
-srsly==2.5.1
-sse-starlette==3.0.2
-stanio==0.5.1
-starlette==0.47.3
-statsmodels==0.14.5
-stringzilla==3.12.6
-stumpy==1.13.0
-sympy==1.13.3
-tables==3.10.2
-tabulate==0.9.0
-tbb==2022.2.0
-tblib==3.1.0
-tcmlib==1.4.0
-tenacity==8.5.0
-tensorboard==2.19.0
-tensorboard-data-server==0.7.2
-tensorflow==2.19.0
-tensorflow-datasets==4.9.9
-tensorflow-hub==0.16.1
-tensorflow-metadata==1.17.2
-tensorflow-probability==0.25.0
-tensorflow-text==2.19.0
-tensorflow_decision_forests==1.12.0
-tensorstore==0.1.76
-termcolor==3.1.0
-terminado==0.18.1
-text-unidecode==1.3
-textblob==0.19.0
-tf-slim==1.1.0
-tf_keras==2.19.0
-thinc==8.3.6
-threadpoolctl==3.6.0
-tifffile==2025.8.28
 tiktoken==0.11.0
-timm==1.0.19
-tinycss2==1.4.0
 tokenizers==0.22.0
-toml==0.10.2
-tomlkit==0.13.3
-toolz==0.12.1
 torch==2.8.0+cu126
 torch-geometric==2.6.1
 torchao==0.10.0
@@ -611,61 +27,7 @@ torchdata==0.11.0
 torchsummary==1.5.1
 torchtune==0.6.1
 torchvision==0.23.0+cu126
-tornado==6.4.2
 tqdm==4.67.1
-traitlets==5.7.1
-traittypes==0.2.1
 transformers==4.56.1
-treelite==4.4.1
-treescope==0.1.10
-triton==3.4.0
-tsfresh==0.21.1
-tweepy==4.16.0
-typeguard==4.4.4
-typer==0.17.3
-types-python-dateutil==2.9.0.20250822
-types-pytz==2025.2.0.20250809
-types-setuptools==80.9.0.20250822
-typing-inspection==0.4.1
-typing_extensions==4.15.0
-tzdata==2025.2
-tzlocal==5.3.1
-uc-micro-py==1.0.3
-ucx-py-cu12==0.44.0
-ucxx-cu12==0.44.0
-umap-learn==0.5.9.post2
-umf==0.11.0
-uri-template==1.3.0
-uritemplate==4.2.0
 urllib3==2.5.0
 uvicorn==0.35.0
-vega-datasets==0.9.0
-wadllib==1.3.6
-wandb==0.21.3
-wasabi==1.1.3
-watchdog==6.0.0
-wcwidth==0.2.13
-weasel==0.4.1
-webcolors==24.11.1
-webencodings==0.5.1
-websocket-client==1.8.0
-websockets==15.0.1
-Werkzeug==3.1.3
-wheel==0.45.1
-widgetsnbextension==3.6.10
-wordcloud==1.9.4
-wrapt==1.17.3
-wurlitzer==3.1.1
-xarray==2025.9.0
-xarray-einstats==0.9.1
-xgboost==3.0.4
-xlrd==2.0.2
-xxhash==3.5.0
-xyzservices==2025.4.0
-yarl==1.20.1
-ydf==0.13.0
-yellowbrick==1.5
-yfinance==0.2.65
-zict==3.0.0
-zipp==3.23.0
-zstandard==0.24.0

src/base_model.py

@@ -43,9 +43,9 @@ def load_model():
             device_map=constants.DEVICE,
         )
     """
-    olf_encoder = torch.load(constants.ENCODER_SMALL_GRAPH_PATH, weights_only=False).to(constants.DEVICE)
+    olf_encoder = torch.jit.load(constants.ENCODER_SMALL_GRAPH_PATH).to(constants.DEVICE)
     olf_encoder.eval()
-    graph_model = torch.load(constants.OVLE_SMALL_GRAPH_PATH, weights_only=False).to(constants.DEVICE)
+    graph_model = torch.jit.load(constants.OVLE_SMALL_GRAPH_PATH).to(constants.DEVICE)
     graph_model.eval()
 
     return clip_model, olf_encoder, graph_model

src/graph_model.py

@@ -40,9 +40,9 @@ def load_model():
             device_map=constants.DEVICE,
         )
     """
-    olf_encoder = torch.load(constants.ENCODER_SMALL_BASE_PATH, weights_only=False).to(constants.DEVICE)
+    olf_encoder = torch.jit.load(constants.ENCODER_SMALL_BASE_PATH).to(constants.DEVICE)
     olf_encoder.eval()
-    graph_model = torch.load(constants.OVLE_SMALL_BASE_PATH, weights_only=False).to(constants.DEVICE)
+    graph_model = torch.jit.load(constants.OVLE_SMALL_BASE_PATH).to(constants.DEVICE)
     graph_model.eval()
 
     return clip_model, olf_encoder, graph_model