Skip to content

Instantly share code, notes, and snippets.

@negedng
Last active November 11, 2020 14:25
Show Gist options
  • Save negedng/a779eaf30dc5e4be079d6d5649751f04 to your computer and use it in GitHub Desktop.
Save negedng/a779eaf30dc5e4be079d6d5649751f04 to your computer and use it in GitHub Desktop.
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
model = AutoModel.from_pretrained("bert-base-cased")
inputs = tokenizer("Hello world!", return_tensors="pt")
outputs = model(**inputs)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment