How to use chandar-lab/NeoBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="chandar-lab/NeoBERT", trust_remote_code=True)
# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("chandar-lab/NeoBERT", trust_remote_code=True, device_map="auto")
hello @marinapLC are you planning on publishing this branch soon ? :)
Hi Charlotte, sorry I dropped it, I would love to get it published.
· Sign up or log in to comment
