towhee.serve.triton.triton_config_builder.TritonModelConfigBuilder

class towhee.serve.triton.triton_config_builder.TritonModelConfigBuilder(model_spec: Dict)[source]

Bases: object

This class is used to generate a triton model config file.

Methods

get_input_schema

get_output_schema

__init__(model_spec: Dict)[source]