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