minor fafb6d9
Francesco Calisto commited on
How to use feynman-integrals-nn/topbox-3layers with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("feynman-integrals-nn/topbox-3layers", device_map="auto")