How to use dmis-lab/UltraBERT-380M with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="dmis-lab/UltraBERT-380M", trust_remote_code=True)
# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("dmis-lab/UltraBERT-380M", trust_remote_code=True, device_map="auto")
The community tab is the place to discuss and collaborate with the HF community!
