How to use benjaminchen22/benchmark-model2 with Transformers:
# Load model directly from transformers import CustomHFModel2 model = CustomHFModel2.from_pretrained("benjaminchen22/benchmark-model2", dtype="auto")