qi\RdZddlmZddlmZddlmZeGddeZdgZy)z( Image/Text processor class for CLIPSeg )ProcessorMixin) BatchEncoding)auto_docstringc2eZdZdfd ZeddZxZS)CLIPSegProcessorc &t|||y)N)super__init__)selfimage_processor tokenizerkwargs __class__s `/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/clipseg/processing_clipseg.pyr zCLIPSegProcessor.__init__s )4c || | td| | td|j|jfd|jji|}||j|fd|i|d}||j |fd|i|d}||j |fd|i|d} || j j d}|S|| j d<|S|S|d j i}|Sttd i | S) a visual_prompt (`PIL.Image.Image`, `np.ndarray`, `torch.Tensor`, `list[PIL.Image.Image]`, `list[np.ndarray]`, `list[torch.Tensor]`): The visual prompt image or batch of images to be prepared. Each visual prompt image can be a PIL image, NumPy array or PyTorch tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a number of channels, H and W are image height and width. Returns: [`BatchEncoding`]: A [`BatchEncoding`] with the following fields: - **input_ids** -- List of token ids to be fed to a model. Returned when `text` is not `None`. - **attention_mask** -- List of indices specifying which tokens should be attended to by the model (when `return_attention_mask=True` or if *"attention_mask"* is in `self.model_input_names` and if `text` is not `None`). - **pixel_values** -- Pixel values to be fed to a model. Returned when `images` is not `None`. z9You have to specify either text, visual prompt or images.zMYou have to specify exactly one type of prompt. Either text or visual prompt.tokenizer_init_kwargsreturn_tensors text_kwargs images_kwargs) pixel_valuesconditional_pixel_valuesrr)data tensor_type) ValueError _merge_kwargsvalid_processor_kwargsr init_kwargsr rrdict) r textimages visual_promptrr output_kwargsencodingprompt_featuresimage_featuress r__call__zCLIPSegProcessor.__call__s" j][hMijH  $2d22.<@Mo@^O  1T11'59F9WN  $); . ; ;,;,H,HHO  &"4'5'B'BH^ $O  O  &*O,H,HHO d&<^&<.Y Yr)NN)NNNN)__name__ __module__ __qualname__r rr( __classcell__)rs@rrrs58Z8ZrrN) __doc__processing_utilsrtokenization_utils_baserutilsrr__all__rrrr2s>/4#=Z~=Z=Z@  r