How to use toind/custom-model with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("toind/custom-model", dtype="auto")