






"""simple docstring"""





import unittest

from diffusers import FlaxAutoencoderKL
from diffusers.utils import is_flax_available
from diffusers.utils.testing_utils import require_flax

from .test_modeling_common_flax import FlaxModelTesterMixin


if is_flax_available():
      import jax


@require_flax
class      _lowercase (     __a		,  unittest.TestCase						):
      _a   :     Optional[Any]								=   FlaxAutoencoderKL


      @property
      def _UpperCamelCase						(    self				:	int						):





            """simple docstring"""







            __snake_case      :    List[str]  			=4
            __snake_case      :    int  			=3
            __snake_case      :    List[Any]  			=(3_2, 3_2)

            __snake_case      :    Union[str, Any]  			=jax.random.PRNGKey(0						)
            __snake_case      :    Dict  			=jax.random.uniform(lowerCAmelCase_							,   ((batch_size, num_channels) + sizes)						)

            return {"sample": image, "prng_key": prng_key}







      def _UpperCamelCase						(    self				:	List[str]						):





            """simple docstring"""







            __snake_case      :    List[str]  			={
                '''block_out_channels''': [3_2, 6_4],
                '''in_channels''': 3,
                '''out_channels''': 3,
                '''down_block_types''': ['''DownEncoderBlock2D''', '''DownEncoderBlock2D'''],
                '''up_block_types''': ['''UpDecoderBlock2D''', '''UpDecoderBlock2D'''],
                '''latent_channels''': 4,
            }
            __snake_case      :    Tuple  			=self.dummy_input
            return init_dict, inputs_dict










"""simple docstring"""





from typing import TYPE_CHECKING

from ...utils import _LazyModule


UpperCamelCase_	:				Union[str, Any]										= {"""tokenization_wav2vec2_phoneme""": ["""Wav2Vec2PhonemeCTCTokenizer"""]}


if TYPE_CHECKING:
    from .tokenization_wavaveca_phoneme import WavaVecaPhonemeCTCTokenizer
else:
    import sys

    UpperCamelCase_	:				List[Any]										= _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)


