






"""simple docstring"""



import copy

from ...configuration_utils import PretrainedConfig
from ...utils import logging
from ..auto.configuration_auto import AutoConfig


_snake_case							:   Union[str, Any]	  =       logging.get_logger(__name__)

_snake_case							:   Tuple	  =       {
    "facebook/musicgen-small": "https://huggingface.co/facebook/musicgen-small/resolve/main/config.json",
    # See all Musicgen models at https://huggingface.co/models?filter=musicgen
}




class 							__lowercase(							lowercase__	):
						lowerCAmelCase_		:				Union[str, Any]=							"""musicgen_decoder"""
						lowerCAmelCase_		:				str=							["""past_key_values"""]






						def __init__(       self     ,    _SCREAMING_SNAKE_CASE=20_48     ,    _SCREAMING_SNAKE_CASE=20_48     ,    _SCREAMING_SNAKE_CASE=24     ,    _SCREAMING_SNAKE_CASE=40_96     ,    _SCREAMING_SNAKE_CASE=16     ,    _SCREAMING_SNAKE_CASE=0.0     ,    _SCREAMING_SNAKE_CASE=True     ,    _SCREAMING_SNAKE_CASE="gelu"     ,    _SCREAMING_SNAKE_CASE=10_24     ,    _SCREAMING_SNAKE_CASE=0.1     ,    _SCREAMING_SNAKE_CASE=0.0     ,    _SCREAMING_SNAKE_CASE=0.0     ,    _SCREAMING_SNAKE_CASE=0.02     ,    _SCREAMING_SNAKE_CASE=False     ,    _SCREAMING_SNAKE_CASE=4     ,    _SCREAMING_SNAKE_CASE=20_48     ,    _SCREAMING_SNAKE_CASE=20_48     ,    _SCREAMING_SNAKE_CASE=None     ,    _SCREAMING_SNAKE_CASE=False     ,    **_SCREAMING_SNAKE_CASE     ,    )      ->	List[Any]:




											"""simple docstring"""





											lowerCamelCase__					=vocab_size
											lowerCamelCase__					=max_position_embeddings
											lowerCamelCase__					=hidden_size
											lowerCamelCase__					=ffn_dim
											lowerCamelCase__					=num_hidden_layers
											lowerCamelCase__					=num_attention_heads
											lowerCamelCase__					=dropout
											lowerCamelCase__					=attention_dropout
											lowerCamelCase__					=activation_dropout
											lowerCamelCase__					=activation_function
											lowerCamelCase__					=initializer_factor
											lowerCamelCase__					=layerdrop
											lowerCamelCase__					=use_cache
											lowerCamelCase__					=scale_embedding  # scale factor will be sqrt(d_model) if True
											lowerCamelCase__					=num_codebooks
											super().__init__(
											    pad_token_id=_SCREAMING_SNAKE_CASE     ,    bos_token_id=_SCREAMING_SNAKE_CASE     ,    eos_token_id=_SCREAMING_SNAKE_CASE     ,    tie_word_embeddings=_SCREAMING_SNAKE_CASE     ,    **_SCREAMING_SNAKE_CASE     ,    )




class 							__lowercase(							lowercase__	):
						lowerCAmelCase_		:				List[str]=							"""musicgen"""
						lowerCAmelCase_		:				Optional[Any]=							True






						def __init__(       self     ,    **_SCREAMING_SNAKE_CASE)      ->	Tuple:




											"""simple docstring"""





											super().__init__(**_SCREAMING_SNAKE_CASE)
											if "text_encoder" not in kwargs or "audio_encoder" not in kwargs or "decoder" not in kwargs:
																raise ValueError('Config has to be initialized with text_encoder, audio_encoder and decoder config')

											lowerCamelCase__					=kwargs.pop('text_encoder')
											lowerCamelCase__					=text_encoder_config.pop('model_type')

											lowerCamelCase__					=kwargs.pop('audio_encoder')
											lowerCamelCase__					=audio_encoder_config.pop('model_type')

											lowerCamelCase__					=kwargs.pop('decoder')

											lowerCamelCase__					=AutoConfig.for_model(_SCREAMING_SNAKE_CASE     ,    **_SCREAMING_SNAKE_CASE)
											lowerCamelCase__					=AutoConfig.for_model(_SCREAMING_SNAKE_CASE     ,    **_SCREAMING_SNAKE_CASE)
											lowerCamelCase__					=MusicgenDecoderConfig(**_SCREAMING_SNAKE_CASE)
											lowerCamelCase__					=True






						@classmethod
						def 		lowerCAmelCase (       cls     ,    _SCREAMING_SNAKE_CASE     ,    _SCREAMING_SNAKE_CASE     ,    _SCREAMING_SNAKE_CASE     ,    **_SCREAMING_SNAKE_CASE     ,    )      ->	Tuple:




											"""simple docstring"""





											return cls(
											    text_encoder=text_encoder_config.to_dict()     ,    audio_encoder=audio_encoder_config.to_dict()     ,    decoder=decoder_config.to_dict()     ,    **_SCREAMING_SNAKE_CASE     ,    )






						def 		lowerCAmelCase (       self)      ->	Any:




											"""simple docstring"""





											lowerCamelCase__					=copy.deepcopy(self.__dict__)
											lowerCamelCase__					=self.text_encoder.to_dict()
											lowerCamelCase__					=self.audio_encoder.to_dict()
											lowerCamelCase__					=self.decoder.to_dict()
											lowerCamelCase__					=self.__class__.model_type
											return output






from __future__ import annotations

from collections.abc import Iterable
from dataclasses import dataclass
from typing import Any



class  _a  :






							"""simple docstring"""

							snake_case_							=    ("_front", "_back", "_len")



							@dataclass
							class  _a  :






														"""simple docstring"""

														snake_case_							=    None
														snake_case_							=    None
														snake_case_							=    None







							class  _a  :






														"""simple docstring"""

														snake_case_							=    ("_cur",)

														def __init__(   self					:  Tuple				,						a					:  Deque._Node | None     )						->None:
																				SCREAMING_SNAKE_CASE__  :					Union[str, Any] =       cur

														def __iter__(   self					:  Dict     )						->Deque._Iterator:
																				return self






														def   A_	(   self					:  int     )						->Any:
																				if self._cur is None:
																										# finished iterating
																										raise StopIteration
																				SCREAMING_SNAKE_CASE__  :					Tuple =       self._cur.val
																				SCREAMING_SNAKE_CASE__  :					int =       self._cur.next_node

																				return val







							def __init__(   self					:  str				,						a					:  Iterable[Any] | None = None     )						->None:
													SCREAMING_SNAKE_CASE__  :					Any =       None
													SCREAMING_SNAKE_CASE__  :					Any =       None
													SCREAMING_SNAKE_CASE__  :					int =       0

													if iterable is not None:
																			# append every value to the deque
																			for val in iterable:
																									self.append(a     )

							def   A_	(   self					:  Tuple				,						a					:  Any     )						->None:
													SCREAMING_SNAKE_CASE__  :					Optional[Any] =       self._Node(a				,						a				,						a     )
													if self.is_empty():
																			# front = back
																			SCREAMING_SNAKE_CASE__  :					int =       node
																			SCREAMING_SNAKE_CASE__  :					Dict =       1
													else:
																			# connect nodes
																			SCREAMING_SNAKE_CASE__  :					Optional[int] =       node
																			SCREAMING_SNAKE_CASE__  :					Any =       self._back
																			SCREAMING_SNAKE_CASE__  :					Tuple =       node  # assign new back to the new node

																			self._len += 1

																			# make sure there were no errors
																			assert not self.is_empty(), "Error on appending value."

							def   A_	(   self					:  List[str]				,						a					:  Any     )						->None:
													SCREAMING_SNAKE_CASE__  :					List[Any] =       self._Node(a				,						a				,						a     )
													if self.is_empty():
																			# front = back
																			SCREAMING_SNAKE_CASE__  :					str =       node
																			SCREAMING_SNAKE_CASE__  :					Dict =       1
													else:
																			# connect nodes
																			SCREAMING_SNAKE_CASE__  :					Any =       self._front
																			SCREAMING_SNAKE_CASE__  :					Tuple =       node
																			SCREAMING_SNAKE_CASE__  :					int =       node  # assign new front to the new node

																			self._len += 1

																			# make sure there were no errors
																			assert not self.is_empty(), "Error on appending value."

							def   A_	(   self					:  Dict				,						a					:  Iterable[Any]     )						->None:
													for val in iterable:
																			self.append(a     )

							def   A_	(   self					:  Tuple				,						a					:  Iterable[Any]     )						->None:
													for val in iterable:
																			self.appendleft(a     )

							def   A_	(   self					:  Any     )						->Any:
													assert not self.is_empty(), "Deque is empty."

													SCREAMING_SNAKE_CASE__  :					List[Any] =       self._back
													SCREAMING_SNAKE_CASE__  :					Optional[int] =       self._back.prev_node  # set new back
													# drop the last node - python will deallocate memory automatically
													SCREAMING_SNAKE_CASE__  :					Union[str, Any] =       None

													self._len -= 1

													return topop.val

							def   A_	(   self					:  Any     )						->Any:
													assert not self.is_empty(), "Deque is empty."

													SCREAMING_SNAKE_CASE__  :					Any =       self._front
													SCREAMING_SNAKE_CASE__  :					Dict =       self._front.next_node  # set new front and drop the first node
													SCREAMING_SNAKE_CASE__  :					Dict =       None

													self._len -= 1

													return topop.val

							def   A_	(   self					:  str     )						->bool:
													return self._front is None

							def __len__(   self					:  List[str]     )						->int:
													return self._len

							def __eq__(   self					:  List[Any]				,						a					:  object     )						->bool:

													if not isinstance(a				,						a     ):
																			return NotImplemented

													SCREAMING_SNAKE_CASE__  :					List[Any] =       self._front
													SCREAMING_SNAKE_CASE__  :					Union[str, Any] =       other._front

													# if the length of the dequeues are not the same, they are not equal
													if len(self     ) != len(a     ):
																			return False

													while me is not None and oth is not None:
																			# compare every value
																			if me.val != oth.val:
																									return False
																			SCREAMING_SNAKE_CASE__  :					Any =       me.next_node
																			SCREAMING_SNAKE_CASE__  :					Any =       oth.next_node

													return True

							def __iter__(   self					:  Any     )						->Deque._Iterator:
													return Deque._Iterator(self._front     )






							def __repr__(   self					:  Any     )						->str:
													SCREAMING_SNAKE_CASE__  :					Union[str, Any] =       []
													SCREAMING_SNAKE_CASE__  :					int =       self._front
													while aux is not None:
																			# append the values in a list to display
																			values_list.append(aux.val     )
																			SCREAMING_SNAKE_CASE__  :					Union[str, Any] =       aux.next_node

													return f"""[{', '.join(repr(a     ) for val in values_list     )}]"""


if __name__ == "__main__":
				import doctest

				doctest.testmod()