allenai / allenai/scibert

AttributeError while using SciBERT

Open
#88 3 comments 0 reactions 0 assignees View on GitHub
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.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.