AttributeError while using SciBERT
Open
- Dominant language
- Python
- Stars
- 1.7k
- Forks
- 232
- PR merge metrics
- No merged PRs in 30d
Description
``
import tensorflow as tf
from transformers import * #BertTokenizer, TFBertModel
tokenizer = AutoTokenizer.from_pretrained('allenai/scibert_scivocab_uncased')
model = AutoModel.from_pretrained('allenai/scibert_scivocab_uncased')
input_ids = tf.constant(tokenizer.encode("My dog is cute"))[None, :] # Batch size 1
outputs = model(input_ids)
last_hidden_states = outputs[0]
print(last_hidden_states)
``
I'm getting following error:
input_shape = input_ids.size()
AttributeError: 'tensorflow.python.framework.ops.EagerTensor' object has no attribute 'size'
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.