o
    ॵi                     @   sv   d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ ejejejdG dd	 d	e	Zd
S )    )AnyDictUnion)	Pipelines)GLM130bForTextGeneration)Pipeline)	PIPELINES)Tasks)	group_keymodule_namec                   @   sR   e Zd Zdeeef fddZdedeeef fddZ	deeef fdd	Z
d
S )GLM130bTextGenerationPipelinemodelc                 O   s    t |tr	t|n|}|| _d S N)
isinstancestrr   r   )selfr   argskwargs r   m/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/pipelines/nlp/glm130b_text_generation_pipeline.py__init__   s   
z&GLM130bTextGenerationPipeline.__init__inputreturnc                 K   s
   |  |S r   )r   )r   r   forward_paramsr   r   r   __call__   s   
z&GLM130bTextGenerationPipeline.__call__c                 K   s   |S )z(This method will not be called.
        r   )r   r   r   r   r   r   postprocess   s   z)GLM130bTextGenerationPipeline.postprocessN)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r   r   r      s    r   N)typingr   r   r   modelscope.metainfor   modelscope.models.nlpr   modelscope.pipelines.baser   modelscope.pipelines.builderr   modelscope.utils.constantr	   register_moduletext_generationglm130b_text_generationr   r   r   r   r   <module>   s   