diff --git a/tests/constraints.txt b/tests/constraints.txt index f27a78153f9..c108e1dde6c 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -26,3 +26,4 @@ pytest-timeout==2.2.0 jax<=0.4.14 jaxlib<=0.4.14 torch>=1.13,<2.3 +kornia==0.7.0 diff --git a/tests/model_hub_tests/pytorch/hf_transformers_models b/tests/model_hub_tests/pytorch/hf_transformers_models index f2596873cac..775797f1b2c 100644 --- a/tests/model_hub_tests/pytorch/hf_transformers_models +++ b/tests/model_hub_tests/pytorch/hf_transformers_models @@ -74,7 +74,7 @@ facebook/encodec_24khz,encodec facebook/esm2_t6_8M_UR50D,esm facebook/flava-full,flava,xfail,Tracing problem facebook/flava-image-codebook,flava_image_codebook,skip,Load problem -facebook/levit-128S,levit,xfail,Tracing problem +facebook/levit-128S,levit facebook/m2m100_418M,m2m_100 facebook/mask2former-swin-base-coco-panoptic,mask2former facebook/maskformer-swin-base-coco,maskformer @@ -367,7 +367,7 @@ thomwolf/vqgan_imagenet_f16_1024,vqgan_model,skip,Load problem thu-ml/zh-clip-vit-roberta-large-patch14,zhclip,skip,Load problem tifa-benchmark/promptcap-coco-vqa,ofa,skip,Load problem tli8hf/robertabase_snli,transformerfornli,skip,Load problem -transfo-xl-wt103,transfo-xl,xfail,Unsupported op aten::index_copy_ +# transfo-xl/transfo-xl-wt103,transfo-xl - deprecated by transformers due to security vulnerability, not inferable in latest transformers transZ/BART_shared_clean,shared_bart,skip,Load problem transZ/BART_shared_v2,shared_bart_v2,skip,Load problem transZ/misecom,misecom,skip,Load problem diff --git a/tests/model_hub_tests/pytorch/requirements.txt b/tests/model_hub_tests/pytorch/requirements.txt index a0c8d326c5b..7063bc75849 100644 --- a/tests/model_hub_tests/pytorch/requirements.txt +++ b/tests/model_hub_tests/pytorch/requirements.txt @@ -24,3 +24,4 @@ torchvision transformers wheel PyYAML +kornia diff --git a/tests/model_hub_tests/pytorch/test_gfpgan.py b/tests/model_hub_tests/pytorch/test_gfpgan.py index a7d8e9374f6..9c123943c28 100644 --- a/tests/model_hub_tests/pytorch/test_gfpgan.py +++ b/tests/model_hub_tests/pytorch/test_gfpgan.py @@ -8,6 +8,8 @@ import tempfile import pytest import torch +import torchvision +from packaging import version from torch_utils import TestTorchConvertModel from openvino import convert_model @@ -76,6 +78,7 @@ class TestGFPGANConvertModel(TestTorchConvertModel): # remove all downloaded files from cache self.repo_dir.cleanup() - @pytest.mark.nightly + @pytest.mark.skipif(version.parse(torchvision.__version__) >= version.parse("0.17"), + reason="torchvision==0.17 have removed module torchvision.transforms.functional_tensor which is required by GFPGAN") def test_convert_model(self, ie_device): self.run("GFPGAN", None, ie_device)