Update quantized model

openvinotoolkit · Oct 6, 2024 · 1fef4b3 · 1fef4b3
1 parent 9a01f9b
commit 1fef4b3
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/src/datumaro/components/algorithms/hash_key_inference/base.py b/src/datumaro/components/algorithms/hash_key_inference/base.py
@@ -21,13 +21,13 @@ def __init__(self, *datasets: Sequence[Dataset]) -> None:
     @property
     def model(self):
         if self._model is None:
-            self._model = explorer.ExplorerLauncher(model_name="clip_visual_vit_l_14_336px")
+            self._model = explorer.ExplorerLauncher(model_name="clip_visual_vit_l_14_336px_int8")
         return self._model
 
     @property
     def text_model(self):
         if self._text_model is None:
-            self._text_model = explorer.ExplorerLauncher(model_name="clip_text_vit_l_14_336px")
+            self._text_model = explorer.ExplorerLauncher(model_name="clip_text_vit_l_14_336px_int8")
         return self._text_model
 
     def _compute_hash_key(self, datasets, datasets_to_infer):

diff --git a/src/datumaro/plugins/openvino_plugin/launcher.py b/src/datumaro/plugins/openvino_plugin/launcher.py
@@ -92,8 +92,8 @@ class BuiltinOpenvinoModelInfo(OpenvinoModelInfo):
     downloadable_models = {
         "clip_text_ViT-B_32",
         "clip_visual_ViT-B_32",
-        "clip_visual_vit_l_14_336px",
-        "clip_text_vit_l_14_336px",
+        "clip_visual_vit_l_14_336px_int8",
+        "clip_text_vit_l_14_336px_int8",
         "googlenet-v4-tf",
     }
 

diff --git a/...amples/clip_text_vit_l_14_336px_interp.py → ...s/clip_text_vit_l_14_336px_int8_interp.py b/...amples/clip_text_vit_l_14_336px_interp.py → ...s/clip_text_vit_l_14_336px_int8_interp.py
diff --git a/...ples/clip_visual_vit_l_14_336px_interp.py → ...clip_visual_vit_l_14_336px_int8_interp.py b/...ples/clip_visual_vit_l_14_336px_interp.py → ...clip_visual_vit_l_14_336px_int8_interp.py