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

[PT FE] Fix xmod model test (#20682)

Browse Source
This commit is contained in:
Maxim Vafin 2023-10-25 08:00:34 +02:00 committed by GitHub
parent 30260e3c7c
commit 46f46c6cc6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/model_hub_tests/torch_tests/test_hf_transformers.py
View File

@ -294,14 +294,11 @@ class TestTransformersModel(TestConvertModel):
@pytest.mark.parametrize("name,type", [("allenai/led-base-16384", "led"), @pytest.mark.parametrize("name,type", [("allenai/led-base-16384", "led"),
("bert-base-uncased", "bert"), ("bert-base-uncased", "bert"),
("facebook/bart-large-mnli", "bart"),
("google/flan-t5-base", "t5"), ("google/flan-t5-base", "t5"),
("google/tapas-large-finetuned-wtq", "tapas"), ("google/tapas-large-finetuned-wtq", "tapas"),
("gpt2", "gpt2"), ("gpt2", "gpt2"),
("openai/clip-vit-large-patch14", "clip"), ("openai/clip-vit-large-patch14", "clip"),
("RWKV/rwkv-4-169m-pile", "rwkv"), ("facebook/xmod-base","xmod")
("microsoft/layoutlmv3-base", "layoutlmv3"),
("microsoft/xprophetnet-large-wiki100-cased", "xlm-prophetnet"),
]) ])
@pytest.mark.precommit @pytest.mark.precommit
def test_convert_model_precommit(self, name, type, ie_device): def test_convert_model_precommit(self, name, type, ie_device):