add model eef44fc
ydshieh commited on
How to use ydshieh/pt_custom_model_example with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("ydshieh/pt_custom_model_example", trust_remote_code=True, dtype="auto")