Method __init__

demo/HuggingFace/T5/export.py:282–284

(self, model, network_metadata)

class T5EncoderTRTEngine(TRTEngineFile):

    def __init__(self, model, network_metadata):
        super().__init__(model, T5EncoderConverter, network_metadata)
        self.max_trt_workspace = T5ModelTRTConfig.MAX_ENCODER_WORKSPACE_MB[network_metadata.variant]

    def get_network_definition(self, network_definition):
        return add_extra_fp32(network_definition)