How to use optimum/legal-bert-base-uncased-neuronx with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("optimum/legal-bert-base-uncased-neuronx") model = AutoModelForPreTraining.from_pretrained("optimum/legal-bert-base-uncased-neuronx", device_map="auto")
