How to use dar-tau/toy-autoencoder-2L-1L with Transformers:
# Load model directly from transformers import ToyAutoEncoder model = ToyAutoEncoder.from_pretrained("dar-tau/toy-autoencoder-2L-1L", device_map="auto")
