Transformers How to use simon-mellergaard/fine_tune_e2e with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("simon-mellergaard/fine_tune_e2e", device_map="auto") 
