




from ...utils import is_note_seq_available, is_transformers_available, is_torch_available
from ...utils import OptionalDependencyNotAvailable


try:
				if not (is_transformers_available() and is_torch_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ...utils.dummy_torch_and_transformers_objects import *  # noqa F403
else:
				from .notes_encoder import SpectrogramNotesEncoder
				from .continous_encoder import SpectrogramContEncoder
				from .pipeline_spectrogram_diffusion import (
				    SpectrogramContEncoder,
				    SpectrogramDiffusionPipeline,
				    TaFilmDecoder,
				)

try:
				if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				from ...utils.dummy_transformers_and_torch_and_note_seq_objects import *  # noqa F403
else:
				from .midi_utils import MidiProcessor





import jax.numpy as jnp

from ...utils import logging
from ..ta.modeling_flax_ta import FlaxTaEncoderModel, FlaxTaForConditionalGeneration, FlaxTaModel
from .configuration_mta import MTaConfig


a													=			logging.get_logger(__name__)

a													=			"T5Config"




def        _SCREAMING_SNAKE_CASE      (		snake_case			,     snake_case			,     snake_case       )						->				jnp.ndarray:
						_UpperCAmelCase      =			jnp.zeros_like(snake_case       )
						_UpperCAmelCase      =			shifted_input_ids.at[:, 1:].set(input_ids[:, :-1]       )
						_UpperCAmelCase      =			shifted_input_ids.at[:, 0].set(snake_case       )

						_UpperCAmelCase      =			jnp.where(shifted_input_ids == -1_0_0			,     snake_case			,     snake_case       )
						return shifted_input_ids




class       _A				(		__lowercase     ):
		__a					    =       """mt5"""
		__a					    =       MTaConfig




class       _A				(		__lowercase     ):
		__a					    =       """mt5"""
		__a					    =       MTaConfig




class       _A				(		__lowercase     ):
		__a					    =       """mt5"""
		__a					    =       MTaConfig