(self)
| 225 | # Converters # |
| 226 | class BARTDecoderConverter(ModelFileConverter): |
| 227 | def __init__(self): |
| 228 | super().__init__(BARTDecoderTorchFile, BARTDecoderONNXFile, BARTDecoderTRTEngine) |
| 229 | |
| 230 | def torch_to_onnx( |
| 231 | self, output_fpath: str, model: Module, network_metadata: NetworkMetadata |