From c21fd2ea6168bd79924455262101719b19f6ea1d Mon Sep 17 00:00:00 2001 From: Ekaterina Aidova Date: Mon, 30 Oct 2023 22:36:08 +0400 Subject: [PATCH] [PT FE]: fix object has no attribute 'dtype' (#20755) * [PT FE]: fix object has no attribute 'dtype' * add model in tests * update test --- .../python/src/openvino/frontend/pytorch/ts_decoder.py | 2 +- tests/model_hub_tests/torch_tests/hf_transformers_models | 1 + tests/model_hub_tests/torch_tests/test_hf_transformers.py | 7 +++++++ 3 files changed, 9 insertions(+), 1 deletion(-) diff --git a/src/bindings/python/src/openvino/frontend/pytorch/ts_decoder.py b/src/bindings/python/src/openvino/frontend/pytorch/ts_decoder.py index a57393e7638..04259234298 100644 --- a/src/bindings/python/src/openvino/frontend/pytorch/ts_decoder.py +++ b/src/bindings/python/src/openvino/frontend/pytorch/ts_decoder.py @@ -67,7 +67,7 @@ class TorchScriptPythonDecoder (Decoder): preserved_attributes = [] for name, module in model.named_modules(): if hasattr(module, "weight"): - if module.weight is not None and module.weight.dtype in [torch.int8, torch.uint8]: + if module.weight is not None and getattr(module.weight, "dtype", None) in [torch.int8, torch.uint8]: preserved_attributes.append(name) return preserved_attributes diff --git a/tests/model_hub_tests/torch_tests/hf_transformers_models b/tests/model_hub_tests/torch_tests/hf_transformers_models index dd41a18235b..ac1e1a8d510 100644 --- a/tests/model_hub_tests/torch_tests/hf_transformers_models +++ b/tests/model_hub_tests/torch_tests/hf_transformers_models @@ -186,6 +186,7 @@ jambran/depression-classification,DepressionDetection,skip,Load problem Jellywibble/dalio-reward-charlie-v1,reward-model,skip,Load problem JonasGeiping/crammed-bert-legacy,crammedBERT,skip,Load problem jonatasgrosman/wav2vec2-large-xlsr-53-english,wav2vec2,xfail,Unsupported op aten::index_put_ prim::TupleConstruct +facebook/mms-lid-126,wav2vec2 Joqsan/test-my-fnet,my_fnet,skip,Load problem jozhang97/deta-swin-large,deta,skip,Load problem jploski/retnet-mini-shakespeare,retnet,skip,Load problem diff --git a/tests/model_hub_tests/torch_tests/test_hf_transformers.py b/tests/model_hub_tests/torch_tests/test_hf_transformers.py index 0dfef7deaae..d29d9dfc538 100644 --- a/tests/model_hub_tests/torch_tests/test_hf_transformers.py +++ b/tests/model_hub_tests/torch_tests/test_hf_transformers.py @@ -104,6 +104,13 @@ class TestTransformersModel(TestConvertModel): model = VIT_GPT2_Model(model) example = (encoded_input.pixel_values,) + elif "mms-lid" in name: + # mms-lid model config does not have auto_model attribute, only direct loading aviable + from transformers import Wav2Vec2ForSequenceClassification, AutoFeatureExtractor + model = Wav2Vec2ForSequenceClassification.from_pretrained(name, torchscript=True) + processor = AutoFeatureExtractor.from_pretrained(name) + input_values = processor(torch.randn(16000).numpy(), sampling_rate=16_000, return_tensors="pt") + example = {"input_values": input_values.input_values} elif "retribert" in mi.tags: from transformers import RetriBertTokenizer text = "How many cats are there?"