diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/llama_index/embeddings/huggingface_openvino/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/llama_index/embeddings/huggingface_openvino/base.py index 0a164040c63d4eab076447a97a124d2aaaae062e..f12527847e3219eb27cdea3a878aaf8194e5963b 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/llama_index/embeddings/huggingface_openvino/base.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/llama_index/embeddings/huggingface_openvino/base.py @@ -82,9 +82,8 @@ class OpenVINOEmbedding(BaseEmbedding): def class_name(cls) -> str: return "OpenVINOEmbedding" - @classmethod + @staticmethod def create_and_save_openvino_model( - cls, model_name_or_path: str, output_path: str, export_kwargs: Optional[dict] = None, @@ -101,7 +100,7 @@ class OpenVINOEmbedding(BaseEmbedding): export_kwargs = export_kwargs or {} model = OVModelForFeatureExtraction.from_pretrained( - model_name_or_path, export=True, **export_kwargs + model_name_or_path, export=True, compile=False, **export_kwargs ) tokenizer = AutoTokenizer.from_pretrained(model_name_or_path) diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/pyproject.toml b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/pyproject.toml index 72e48ebcab08b9129c80844e95514b1150e67996..c61fbcd732891f8d7cef70df04c33108cc129edf 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/pyproject.toml +++ b/llama-index-integrations/embeddings/llama-index-embeddings-huggingface-openvino/pyproject.toml @@ -27,7 +27,7 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-embeddings-openvino" readme = "README.md" -version = "0.1.5" +version = "0.1.6" [tool.poetry.dependencies] python = ">=3.8.1,<4.0"