heavycross im 9f20926
Francesco Calisto commited on
How to use feynman-integrals-nn/heavycrossbox with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("feynman-integrals-nn/heavycrossbox", device_map="auto")