AttributeError: 'MolformerModel' object has no attribute 'warn_if_padding_and_no_attention_mask'
#2
by
pamo77
- opened
Hello,
I am running into a problem when trying out the example code (out-of-the-box) shown in the Model Card, i.e.
import torch
from transformers import AutoModel, AutoTokenizer
model = AutoModel.from_pretrained("ibm/MoLFormer-XL-both-10pct", deterministic_eval=True, trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("ibm/MoLFormer-XL-both-10pct", trust_remote_code=True)
smiles = ["Cn1c(=O)c2c(ncn2C)n(C)c1=O", "CC(=O)Oc1ccccc1C(=O)O"]
inputs = tokenizer(smiles, padding=True, return_tensors="pt")
with torch.no_grad():
outputs = model(**inputs)
outputs.pooler_output
I am getting the following error:AttributeError: 'MolformerModel' object has no attribute 'warn_if_padding_and_no_attention_mask'
Is there a workaround for this?
Many thanks in advance.
The full error dump looks like this:
in <module>:7 │
│ │
│ 4 smiles_0 = ["Cn1c(=O)c2c(ncn2C)n(C)c1=O", "CC(=O)Oc1ccccc1C(=O)O"] │
│ 5 inputs_0 = tokenizer_0(smiles_0, padding=True, return_tensors="pt") │
│ 6 with torch.no_grad(): │
│ ❱ 7 │ outputs = model_0(**inputs_0) │
│ 8 outputs.pooler_output │
│ 9 │
│ │
│ /Users/<user>/.pyenv/versions/3.10.11/envs/my-31011-python/lib/python3.10/site-packa │
│ ges/torch/nn/modules/module.py:1501 in _call_impl │
│ │
│ 1498 │ │ if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks │
│ 1499 │ │ │ │ or _global_backward_pre_hooks or _global_backward_hooks │
│ 1500 │ │ │ │ or _global_forward_hooks or _global_forward_pre_hooks): │
│ ❱ 1501 │ │ │ return forward_call(*args, **kwargs) │
│ 1502 │ │ # Do not call functions when jit is used │
│ 1503 │ │ full_backward_hooks, non_full_backward_hooks = [], [] │
│ 1504 │ │ backward_pre_hooks = [] │
│ │
│ /Users/<user>/.cache/huggingface/modules/transformers_modules/ibm/MoLFormer-XL-both- │
│ 10pct/7b12d946c181a37f6012b9dc3b002275de070314/modeling_molformer.py:650 in forward │
│ │
│ 647 │ │ if input_ids is not None and inputs_embeds is not None: │
│ 648 │ │ │ raise ValueError("You cannot specify both input_ids and inputs_embeds at the │
│ 649 │ │ elif input_ids is not None: │
│ ❱ 650 │ │ │ self.warn_if_padding_and_no_attention_mask(input_ids, attention_mask) │
│ 651 │ │ │ input_shape = input_ids.size() │
│ 652 │ │ elif inputs_embeds is not None: │
│ 653 │ │ │ input_shape = inputs_embeds.size()[:-1] │
│ │
│ /Users/<user>/.pyenv/versions/3.10.11/envs/my-31011-python/lib/python3.10/site-packa │
│ ges/torch/nn/modules/module.py:1614 in __getattr__ │
│ │
│ 1611 │ │ │ modules = self.__dict__['_modules'] │
│ 1612 │ │ │ if name in modules: │
│ 1613 │ │ │ │ return modules[name] │
│ ❱ 1614 │ │ raise AttributeError("'{}' object has no attribute '{}'".format( │
│ 1615 │ │ │ type(self).__name__, name)) │
│ 1616 │ │
│ 1617 │ def __setattr__(self, name: str, value: Union[Tensor, 'Module']) -> None:
Try to upgrade the transformers, warn_if_padding_and_no_attention_mask is an attribute of PreTrainedModel