MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / BARTDecoderONNXFile

Class BARTDecoderONNXFile

demo/HuggingFace/BART/export.py:194–196  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

192
193
194class BARTDecoderONNXFile(ONNXModelFile):
195 def __init__(self, model, network_metadata):
196 super().__init__(model, BARTDecoderConverter, network_metadata)
197
198
199# TRT Engine File Encoding #

Callers 2

_setup_enginesMethod · 0.90
torch_to_onnxMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected