Skip to content
Merged
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.24
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