How to use yasinamp/simple_nn_model with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("yasinamp/simple_nn_model", trust_remote_code=True, dtype="auto")