How to use bada/test with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("bada/test") model = AutoModelForPreTraining.from_pretrained("bada/test", device_map="auto")
