natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PT FE] Fix GHA model tests (#23846) 

### Details:
 - *Added `kornia` dependency for lightglue model*
 - *Disable GFPGAN since it doesn't support latest `torchvision`*
 - *Remove `transfo-xl-wt103` as not supported on latest `transformers`*
 - *Remove XFAIL from `facebook/levit-128S`*

### Tickets:
 - *ticket-id*
This commit is contained in:
Maxim Vafin 2024-04-05 08:30:50 +02:00 committed by GitHub
parent c98a861e5b
commit 4ef40f0f0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/constraints.txt
View File

@ -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

4
tests/model_hub_tests/pytorch/hf_transformers_models
View File

@ -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

1
tests/model_hub_tests/pytorch/requirements.txt
View File

@ -24,3 +24,4 @@ torchvision
transformers
wheel
PyYAML
kornia

5
tests/model_hub_tests/pytorch/test_gfpgan.py
View File

@ -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)