How to use goku/trainer_output with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("goku/trainer_output", dtype="auto")