(self, model, network_metadata)
| 255 | class T5DecoderTRTEngine(TRTEngineFile): |
| 256 | |
| 257 | def __init__(self, model, network_metadata): |
| 258 | super().__init__(model, T5DecoderConverter, network_metadata) |
| 259 | self.max_trt_workspace = T5ModelTRTConfig.MAX_DECODER_WORKSPACE_MB[network_metadata.variant] |
| 260 | |
| 261 | |
| 262 | def get_network_definition(self, network_definition): |