Transformers How to use flax-community/gpt2-Cosmos with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForMultipleChoice
tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt2-Cosmos")
model = AutoModelForMultipleChoice.from_pretrained("flax-community/gpt2-Cosmos", device_map="auto") 
