Skip to content

Instantly share code, notes, and snippets.

@prakhar21

prakhar21/transformer_interpet_model_loading.py

Created May 8, 2021 10:56
Show Gist options
  • Save prakhar21/d94def73a45b1fc4d45d318dd1a1fc4f to your computer and use it in GitHub Desktop.
Save prakhar21/d94def73a45b1fc4d45d318dd1a1fc4f to your computer and use it in GitHub Desktop.
Download ZIP
transformer-interpet_model_loading
Raw
transformer_interpet_model_loading.py
from transformers import AutoModelForSequenceClassification, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")
model = AutoModelForSequenceClassification.from_pretrained(
"sampathkethineedi/industry-classification"
)
print (tokenizer, model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment