o ॵi @sddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZd gZejejejd Gd d d e ZdS) )AnyDictN) Pipelines)Model) OutputKeys)Pipeline) PIPELINES)TasksPipelineTemplate) module_namec seZdZdZdeffdd Z  ddeded ed efd d Z  ddeded ed efd dZ ddede d e e effddZ ZS)r a:A pipeline template explain how to define parameters and input and output information. As a rule, the first parameter is the input, followed by the request parameters. The parameter must add type hint information, and set the default value if necessary, for the convenience of use. modelc stjdd|i|dS)zA pipeline template to describe input and output and parameter processing Args: model: A Model instance. r N)super__init__)selfr kwargs __class__r Z/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/pipelines/pipeline_template.pyrszPipelineTemplate.__init__皙?input max_lengthtop_preturncCdS)alPipeline preprocess interface. Args: input (Any): The pipeline input, ref Tasks.task_template TASK_INPUTS. max_length (int, optional): The max_length parameter. Defaults to 1024. top_p (float, optional): The top_p parameter. Defaults to 0.8. Returns: Any: Return result process by forward. Nr rrrrr r r preprocess%zPipelineTemplate.preprocesscCr)a1The forward interface. Args: input (Any): The output of the preprocess. max_length (int, optional): max_length. Defaults to 1024. top_p (float, optional): top_p. Defaults to 0.8. Returns: Any: Return result process by postprocess. Nr rr r rforward5rzPipelineTemplate.forwardNinputspostprocess_param1c Cs0tjtdtjtddtjtddi}|S)a2The postprocess interface. Args: input (Any): The output of the forward. max_length (int, optional): max_length. Defaults to 1024. top_p (float, optional): top_p. Defaults to 0.8. Returns: Any: Return result process by postprocess.  i)rBOXESnpzeros OUTPUT_IMGTEXT_EMBEDDING)rr r!resultr r r postprocessEs zPipelineTemplate.postprocess)rr)N)__name__ __module__ __qualname____doc__rrrintfloatrrstrrr+ __classcell__r r rrr sB    )typingrrnumpyr&modelscope.metainfor!modelscope.models.base.base_modelrmodelscope.outputs.outputsrmodelscope.pipelines.basermodelscope.pipelines.builderrmodelscope.utils.constantr __all__register_module task_templatepipeline_templater r r r rs