How to use flax-community/bigband with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("flax-community/bigband") model = AutoModelForPreTraining.from_pretrained("flax-community/bigband")