How to use DrLonee/scibert-citation-model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="DrLonee/scibert-citation-model")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("DrLonee/scibert-citation-model") model = AutoModelForSequenceClassification.from_pretrained("DrLonee/scibert-citation-model", device_map="auto")
