initial commit from ajitrajasekharan 7a5c2bf
ajit commited on
How to use ajitrajasekharan/biomedical with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="ajitrajasekharan/biomedical") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("ajitrajasekharan/biomedical")
model = AutoModelForMaskedLM.from_pretrained("ajitrajasekharan/biomedical", device_map="auto")