End of training 15084a2 verified
Razvan Vasile commited on
How to use rfvasile/trainer_output with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("rfvasile/trainer_output", device_map="auto")