






"""simple docstring"""







from typing import List, Optional, Union

from ...configuration_utils import PretrainedConfig
from ...utils import logging


lowerCAmelCase_	:    str            =      logging.get_logger(__name__)

lowerCAmelCase_	:    Optional[int]            =      {
    '''huggingface/time-series-transformer-tourism-monthly''': (
        '''https://huggingface.co/huggingface/time-series-transformer-tourism-monthly/resolve/main/config.json'''
    ),
    # See all TimeSeriesTransformer models at https://huggingface.co/models?filter=time_series_transformer
}





class   UpperCamelCase_ (       a_  ):
  _A       :					int				=	'time_series_transformer'
  _A       :					List[Any]				=	{
      'hidden_size': 'd_model',
      'num_attention_heads': 'encoder_attention_heads',
      'num_hidden_layers': 'encoder_layers',
  }



  def __init__(      self ,		snake_case__ = None ,		snake_case__ = None ,		snake_case__ = "student_t" ,		snake_case__ = "nll" ,		snake_case__ = 1 ,		snake_case__ = [1, 2, 3, 4, 5, 6, 7] ,		snake_case__ = "mean" ,		snake_case__ = 0 ,		snake_case__ = 0 ,		snake_case__ = 0 ,		snake_case__ = 0 ,		snake_case__ = None ,		snake_case__ = None ,		snake_case__ = 32 ,		snake_case__ = 32 ,		snake_case__ = 2 ,		snake_case__ = 2 ,		snake_case__ = 2 ,		snake_case__ = 2 ,		snake_case__ = True ,		snake_case__ = "gelu" ,		snake_case__ = 64 ,		snake_case__ = 0.1 ,		snake_case__ = 0.1 ,		snake_case__ = 0.1 ,		snake_case__ = 0.1 ,		snake_case__ = 0.1 ,		snake_case__ = 1_00 ,		snake_case__ = 0.02 ,		snake_case__=True ,		**snake_case__ ,		)			->			Union[str, Any]:



         """simple docstring"""





         UpperCAmelCase	    =			prediction_length
         UpperCAmelCase	    =			context_length or prediction_length
         UpperCAmelCase	    =			distribution_output
         UpperCAmelCase	    =			loss
         UpperCAmelCase	    =			input_size
         UpperCAmelCase	    =			num_time_features
         UpperCAmelCase	    =			lags_sequence
         UpperCAmelCase	    =			scaling
         UpperCAmelCase	    =			num_dynamic_real_features
         UpperCAmelCase	    =			num_static_real_features
         UpperCAmelCase	    =			num_static_categorical_features
         if cardinality and num_static_categorical_features > 0:
                if len(snake_case__				) != num_static_categorical_features:
                       raise ValueError(
                           """The cardinality should be a list of the same length as `num_static_categorical_features`"""				)
                UpperCAmelCase	    =			cardinality
         else:
                UpperCAmelCase	    =			[0]
         if embedding_dimension and num_static_categorical_features > 0:
                if len(snake_case__				) != num_static_categorical_features:
                       raise ValueError(
                           """The embedding dimension should be a list of the same length as `num_static_categorical_features`"""				)
                UpperCAmelCase	    =			embedding_dimension
         else:
                UpperCAmelCase	    =			[min(50 ,		(cat + 1) // 2				) for cat in self.cardinality]
         UpperCAmelCase	    =			num_parallel_samples

         # Transformer architecture configuration
         UpperCAmelCase	    =			input_size * len(snake_case__				) + self._number_of_features
         UpperCAmelCase	    =			d_model
         UpperCAmelCase	    =			encoder_attention_heads
         UpperCAmelCase	    =			decoder_attention_heads
         UpperCAmelCase	    =			encoder_ffn_dim
         UpperCAmelCase	    =			decoder_ffn_dim
         UpperCAmelCase	    =			encoder_layers
         UpperCAmelCase	    =			decoder_layers

         UpperCAmelCase	    =			dropout
         UpperCAmelCase	    =			attention_dropout
         UpperCAmelCase	    =			activation_dropout
         UpperCAmelCase	    =			encoder_layerdrop
         UpperCAmelCase	    =			decoder_layerdrop

         UpperCAmelCase	    =			activation_function
         UpperCAmelCase	    =			init_std

         UpperCAmelCase	    =			use_cache

         super().__init__(is_encoder_decoder=snake_case__ ,		**snake_case__				)



  @property
  def 		UpperCamelCase_				(      self				)			->			int:



         """simple docstring"""





         return (
             sum(self.embedding_dimension				)
             + self.num_dynamic_real_features
             + self.num_time_features
             + self.num_static_real_features
             + self.input_size * 2  # the log1p(abs(loc)) and log(scale) features
         )











"""simple docstring"""







import warnings

from ...processing_utils import ProcessorMixin
from ...tokenization_utils_base import BatchEncoding





class   UpperCamelCase_ (       a_  ):
  _A       :					Optional[int]				=	['image_processor', 'tokenizer']
  _A       :					Optional[Any]				=	'ViTImageProcessor'
  _A       :					Tuple				=	('CLIPTokenizer', 'CLIPTokenizerFast')



  def __init__(      self ,		snake_case__=None ,		snake_case__=None ,		**snake_case__				)			->			Union[str, Any]:



         """simple docstring"""





         UpperCAmelCase	    =			None
         if "feature_extractor" in kwargs:
                warnings.warn(
                    """The `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor`"""
                    """ instead.""" ,		snake_case__ ,		)
                UpperCAmelCase	    =			kwargs.pop("""feature_extractor"""				)

         UpperCAmelCase	    =			image_processor if image_processor is not None else feature_extractor
         if image_processor is None:
                raise ValueError("""You need to specify an `image_processor`."""				)
         if tokenizer is None:
                raise ValueError("""You need to specify a `tokenizer`."""				)

         super().__init__(snake_case__ ,		snake_case__				)



  def __call__(      self ,		snake_case__=None ,		snake_case__=None ,		snake_case__=None ,		snake_case__=None ,		**snake_case__				)			->			List[str]:



         """simple docstring"""





         if text is None and visual_prompt is None and images is None:
                raise ValueError("""You have to specify either text, visual prompt or images."""				)

         if text is not None and visual_prompt is not None:
                raise ValueError("""You have to specify exactly one type of prompt. Either text or visual prompt."""				)

         if text is not None:
                UpperCAmelCase	    =			self.tokenizer(snake_case__ ,		return_tensors=snake_case__ ,		**snake_case__				)

         if visual_prompt is not None:
                UpperCAmelCase	    =			self.image_processor(snake_case__ ,		return_tensors=snake_case__ ,		**snake_case__				)

         if images is not None:
                UpperCAmelCase	    =			self.image_processor(snake_case__ ,		return_tensors=snake_case__ ,		**snake_case__				)

         if visual_prompt is not None and images is not None:
                UpperCAmelCase	    =			{
                    """pixel_values""": image_features.pixel_values,
                    """conditional_pixel_values""": prompt_features.pixel_values,
                }
                return encoding
         elif text is not None and images is not None:
                UpperCAmelCase	    =			image_features.pixel_values
                return encoding
         elif text is not None:
                return encoding
         elif visual_prompt is not None:
                UpperCAmelCase	    =			{
                    """conditional_pixel_values""": prompt_features.pixel_values,
                }
                return encoding
         else:
                return BatchEncoding(data=dict(**snake_case__				) ,		tensor_type=snake_case__				)



  def 		UpperCamelCase_				(      self ,		*snake_case__ ,		**snake_case__				)			->			Tuple:



         """simple docstring"""





         return self.tokenizer.batch_decode(*snake_case__ ,		**snake_case__				)



  def 		UpperCamelCase_				(      self ,		*snake_case__ ,		**snake_case__				)			->			Optional[int]:



         """simple docstring"""





         return self.tokenizer.decode(*snake_case__ ,		**snake_case__				)



  @property
  def 		UpperCamelCase_				(      self				)			->			Tuple:



         """simple docstring"""





         warnings.warn(
             """`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead.""" ,		snake_case__ ,		)
         return self.image_processor_class



  @property
  def 		UpperCamelCase_				(      self				)			->			Any:



         """simple docstring"""





         warnings.warn(
             """`feature_extractor` is deprecated and will be removed in v5. Use `image_processor` instead.""" ,		snake_case__ ,		)
         return self.image_processor
