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

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

@ -250,8 +250,8 @@ class TestTransformersModel(TestConvertModel):
if model is None: if model is None:
from transformers import AutoModel from transformers import AutoModel
model = AutoModel.from_pretrained(name, torchscript=True) model = AutoModel.from_pretrained(name, torchscript=True)
if hasattr(model, "set_default_language"): if hasattr(model, "set_default_language"):
model.set_default_language("en_XX") model.set_default_language("en_XX")
if example is None: if example is None:
if "encodec" in mi.tags: if "encodec" in mi.tags:
example = (torch.randn(1, 1, 100),) example = (torch.randn(1, 1, 100),)
@ -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):