Legal BigBird (ONNX)
This is a fine-tuned version of BigBird-Pegasus for legal document summarization. It has been exported to ONNX for high-performance inference.
Usage
import onnxruntime as ort
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("arirajuns/bigbird-legal-onnx")
session = ort.InferenceSession("model.onnx")
inputs = tokenizer("Your long legal text here...", return_tensors="np")
outputs = session.run(None, dict(inputs))
- Downloads last month
- 7
Model tree for arirajuns/bigbird-legal-onnx
Base model
google/bigbird-pegasus-large-arxiv