towhee.serve.triton.python_model_builder.PyModelBuilder

class towhee.serve.triton.python_model_builder.PyModelBuilder[source]

Bases: object

The base class of python model builder. Subclass of PyModelBuilder should implement the following methods: gen_import, gen_init.

Methods

build

gen_blank_lines

gen_class_name

gen_execute

gen_finalize

gen_imports

gen_initialize