vj ddlmZmZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZd gZejejej Gd d e ZdS) )AnyDictN) Pipelines)Model) OutputKeys)Pipeline) PIPELINES)TasksPipelineTemplate) module_namec eZdZdZdeffd Z ddededed efd Z ddededed efd Z dd ede d e e effdZ xZS)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 >tjdd|i|dS)zA pipeline template to describe input and output and parameter processing Args: model: A Model instance. rN)super__init__)selfrkwargs __class__s v/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/pipelines/pipeline_template.pyrzPipelineTemplate.__init__s- //u//////皙?input max_lengthtop_preturncdS)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. Nrrrrrs r preprocesszPipelineTemplate.preprocess%  rcdS)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. Nrrs rforwardzPipelineTemplate.forward5r!rNinputspostprocess_param1c tjtjdtjtjddtjtjddi}|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%results r postprocesszPipelineTemplate.postprocessEsH  bhqkk  !28B??  %rx4'8'8   r)rr)N)__name__ __module__ __qualname____doc__rrrintfloatr r#strrr0 __classcell__)rs@rr r s0e000000&*"%   "   +.    $#'"     (+    $/3(+7;CH~r)typingrrnumpyr+modelscope.metainfor!modelscope.models.base.base_modelrmodelscope.outputs.outputsrmodelscope.pipelines.basermodelscope.pipelines.builderr modelscope.utils.constantr __all__register_module task_templatepipeline_templater rrrrEs))))))333333111111......222222++++++   Y%@BBBDDDDDxDDBBDDDr