Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion tests/e2e_tests/pipelines/tf_hub/nightly.yml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
universal-sentence-encoder,https://tfhub.dev/google/universal-sentence-encoder/4?tf-hub-format=compressed
imagenet/mobilenet_v1_100_224/classification,https://tfhub.dev/google/imagenet/mobilenet_v1_100_224/classification/5?tf-hub-format=compressed
imagenet/mobilenet_v2_100_224/classification,https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/classification/5?tf-hub-format=compressed
universal-sentence-encoder-multilingual,https://tfhub.dev/google/universal-sentence-encoder-multilingual/3?tf-hub-format=compressed
universal-sentence-encoder-multilingual,https://tfhub.dev/google/universal-sentence-encoder-multilingual/3?tf-hub-format=compressed,skip,ticket 179327
universal-sentence-encoder-large,https://tfhub.dev/google/universal-sentence-encoder-large/5?tf-hub-format=compressed
imagenet/mobilenet_v2_075_224/classification,https://tfhub.dev/google/imagenet/mobilenet_v2_075_224/classification/5?tf-hub-format=compressed
movenet/singlepose/lightning,https://tfhub.dev/google/movenet/singlepose/lightning/4?tf-hub-format=compressed
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
universal-sentence-encoder,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/universal-sentence-encoder/versions/2,xfail,128994 127962 128995 128996 unsupported operations LookupTableFindV2 LookupTableSizeV2 SparseFillEmptyRows StaticRegexReplace StringSplit StringToHashBucketFast
imagenet/mobilenet_v1_100_224/classification,https://www.kaggle.com/models/google/mobilenet-v1/frameworks/tensorFlow2/variations/100-224-classification/versions/2
imagenet/mobilenet_v2_100_224/classification,https://www.kaggle.com/models/google/mobilenet-v2/frameworks/tensorFlow2/variations/100-224-classification/versions/2
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2,skip,ticket 179327
universal-sentence-encoder-large,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/large/versions/2,xfail,128994 127962 128995 128996 unsupported operations LookupTableFindV2 LookupTableSizeV2 SparseFillEmptyRows StaticRegexReplace StringSplit StringToHashBucketFast
imagenet/mobilenet_v2_075_224/classification,https://www.kaggle.com/models/google/mobilenet-v2/frameworks/tensorFlow2/variations/075-224-classification/versions/2
movenet/singlepose/lightning,https://www.kaggle.com/models/google/movenet/frameworks/tensorFlow2/variations/singlepose-lightning/versions/4
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ planet/vision/classifier/planet_v2,https://www.kaggle.com/models/google/planet-v
# TF1 models in .pb format
i3d-rgb,https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/i3d-rgb-tf/rgb.frozen.pb
# Model with SentencePiece tokenizer, use openvino-tokenizers package
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2,skip,ticket 179327
# Hugging Face Models
microsoft/deberta-v3-small,hf_transformers
facebook/convnext-tiny-224,hf_transformers
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
mil-nce/s3d,https://www.kaggle.com/models/deepmind/mil-nce/frameworks/tensorFlow1/variations/s3d/versions/1
# models from precommit_convert_model
yamnet,https://www.kaggle.com/models/google/yamnet/frameworks/tensorFlow2/variations/yamnet/versions/1
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2
universal-sentence-encoder-multilingual,https://www.kaggle.com/models/google/universal-sentence-encoder/frameworks/tensorFlow2/variations/multilingual/versions/2,skip,ticket 179327
movenet/singlepose/lightning,https://www.kaggle.com/models/google/movenet/frameworks/tensorFlow2/variations/singlepose-lightning/versions/4
imagenet/resnet_v2_50/feature_vector,https://www.kaggle.com/models/google/resnet-v2/frameworks/tensorFlow2/variations/50-feature-vector/versions/2
# LookupTableImportV2 is terminating node and is needed for conversion
Expand Down
2 changes: 1 addition & 1 deletion tests/model_hub_tests/tensorflow/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,6 @@ pytest
pytest-html
tensorflow
tensorflow-hub
tensorflow-text
#tensorflow-text # ticket 179327
transformers
tf-sentence-transformers
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import tensorflow.compat.v1 as tf_v1
import tensorflow_hub as hub
# noinspection PyUnresolvedReferences
import tensorflow_text # do not delete, needed for text models
#import tensorflow_text # do not delete, needed for text models. Commended due to ticket 179327
from huggingface_hub import snapshot_download
from models_hub_common.test_convert_model import TestConvertModel
from models_hub_common.utils import get_models_list
Expand Down
2 changes: 1 addition & 1 deletion tests/model_hub_tests/tensorflow/test_tf_read_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import tensorflow as tf
import tensorflow_hub as hub
# noinspection PyUnresolvedReferences
import tensorflow_text # do not delete, needed for text models
#import tensorflow_text # do not delete, needed for text models. Commented due to ticket 179327
from models_hub_common.test_convert_model import TestConvertModel
from models_hub_common.utils import get_models_list
from openvino import Core, PartialShape
Expand Down
8 changes: 4 additions & 4 deletions tests/requirements_pytorch
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ defusedxml==0.7.1

autoawq==0.2.7; platform_system == "Linux" and platform_machine == "x86_64"
# triton is a dependency of autoawq, newer versions lead to TorchFX test failures
triton==3.1.0; platform_system == "Linux" and platform_machine == "x86_64"
triton==3.5.1; platform_system == "Linux" and platform_machine == "x86_64"
auto-gptq==0.7.1; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.12"
# peft is a dependency of auto-gptq, newer versions lead to gptq model loading failures
peft==0.17.1; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.12"
Expand All @@ -30,10 +30,10 @@ facexlib==0.3.0; python_version < "3.12"
librosa==0.10.2; python_version < "3.12"
packaging==24.1
pandas==2.2.3
protobuf==5.28.2
protobuf==5.28.3
pyctcdecode==0.5.0; python_version < "3.12"
sacremoses==0.1.1
sentencepiece==0.2.0
sentencepiece==0.2.1
soundfile==0.12.1
super-image==0.1.7; python_version < "3.12"
timm==1.0.11
Expand All @@ -48,7 +48,7 @@ huggingface-hub==0.25.2
# It will be discussed in the future how to manage versioning of the components properly.
git+https://github.com/huggingface/optimum-intel.git@8ba536cd0a2bf93e9e88408b0048a7695db5be0b; python_version < "3.12"
# set 'export HF_HUB_ENABLE_HF_TRANSFER=1' to benefits from hf_transfer
hf_transfer==0.1.8
hf_transfer==0.1.9

# requirements for specific models
# - hf-tiny-model-private/tiny-random-RoFormerForCausalLM
Expand Down
17 changes: 8 additions & 9 deletions tests/requirements_tensorflow
Original file line number Diff line number Diff line change
Expand Up @@ -3,23 +3,22 @@
# tensorflow-intel 2.18.0 depends on numpy<2.1.0 and >=1.26.0
# tensorflow 2.16.2 depends on numpy<2.0.0 and >=1.26.0; python_version >= "3.12"
numpy==1.26.4; python_version < "3.12" or platform_system == "Darwin" and platform_machine == "x86_64"
numpy==2.0.2; python_version >= "3.12" and (platform_system != "Darwin" or platform_machine != "x86_64")
pytest==7.0.1; python_version < '3.10'
pytest==7.2.0; python_version >= '3.10'
numpy==2.2.6; python_version >= "3.12" and (platform_system != "Darwin" or platform_machine != "x86_64")
pytest==7.2.0
pytest-xdist[psutil]==3.6.1
pytest-html==4.2.0
transformers==4.45.1
# install exact keras version since tensorflow depends and has no upper bound for it
keras==3.8.0
tensorflow==2.18.0; platform_system != "Darwin" or platform_machine != "x86_64"
keras==3.13.0; python_version >= "3.13"
keras==3.12.0; python_version < "3.13"
tensorflow==2.20.0; platform_system != "Darwin" or platform_machine != "x86_64"
tensorflow==2.16.2; platform_system == "Darwin" and platform_machine == "x86_64"
# install explicit version of wrapt to avoid "this __dict__ descriptor does not support '_DictWrapper' objects" error from TensorFlow 2.18
wrapt==1.15.0; python_version >= "3.12"
# tensorflow-text is not available for both Windows and ARM platforms
tensorflow-text==2.18.0; python_version < "3.12" and platform_system == "Linux" and platform_machine == "x86_64"
#tensorflow-text==2.19.0; python_version < "3.13" and platform_system == "Linux" and platform_machine == "x86_64" # ticket 179327
tensorflow-hub==0.16.1
jax==0.5.0; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9"
jax==0.5.0; (platform_system != "Darwin" or platform_machine != "x86_64")
# tensorflow 2.16.2 depends on ml-dtypes~=0.3.1 and jax 0.4.35 depends on ml-dtypes>=0.4.0
jax==0.4.33; (platform_system == "Darwin" and platform_machine == "x86_64") and python_version > "3.9"
jax==0.4.30; python_version <= "3.9"
jax==0.4.33; (platform_system == "Darwin" and platform_machine == "x86_64")
defusedxml==0.7.1
Loading