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