tj\`dZddlmZddlmZddlmZeGddeZdgZdS)z( Image/Text processor class for CLIPSeg )ProcessorMixin) BatchEncoding)auto_docstringc8eZdZdfd ZeddZxZS)CLIPSegProcessorNc Lt||dS)N)super__init__)selfimage_processor tokenizerkwargs __class__s l/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/clipseg/processing_clipseg.pyr zCLIPSegProcessor.__init__s# )44444c |||td||td|j|jfd|jji|}||j|fd|i|d}||j|fd|i|d}||j|fd|i|d} ||| j|jd}|S|| | j|d <|S||S| d |ji}|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`. Nz9You 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][hMijjH  $2d2.<@Mo@^O  1T1'59F9WN  $); . ;,;,HHO  &"4'5'BH^ $O  O  &*O,HHO d&<&<^&<&<.YYY Yr)NN)NNNN)__name__ __module__ __qualname__r rr( __classcell__)rs@rrrse5555558Z8Z8Z^8Z8Z8Z8Z8ZrrN) __doc__processing_utilsrtokenization_utils_baserutilsrr__all__rrrr2s/.....444444######=Z=Z=Z=Z=Z~=Z=Z=Z@  r