How to use 1stub/mlops-basic with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("1stub/mlops-basic", device_map="auto")
