Pre-trained on TCDB 36cf782
Hamed Ghazikhani commited on
How to use ghazikhanihamed/TCDB-BERT with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="ghazikhanihamed/TCDB-BERT") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("ghazikhanihamed/TCDB-BERT")
model = AutoModelForMaskedLM.from_pretrained("ghazikhanihamed/TCDB-BERT", device_map="auto")